dniku / rl-attention

dniku/rl-attention Issues

Visualize attention in callback
Closed 5 years ago1
Make sure that vanilla architecture earns high reward on multiple envs
Closed 5 years ago1
Export trained models, evaluation results and training stats from Colab
Closed 5 years ago2
Implement entropy loss from paper
Updated 5 years ago
Implement entropy loss
Closed 5 years ago1
Save metrics after training
Closed 5 years ago2
Review of attention model
Closed 5 years ago1
Fix non-determinism
Closed 5 years ago1
Implement attention model from LTIAA
Closed 5 years ago
Ensure that agent earns high reward on Colab
Closed 5 years ago1
Implement visualization of attention map
Closed 5 years ago
Make code runnable on Colab
Closed 5 years ago
Specify network architecture in our code
Closed 5 years ago1
Implement skeleton based on stable-baselines
Closed 5 years ago1