hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-08 19:41:38 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions
ray/rllib/models/torch/modules
History
Sven Mika d5604eaba3
[RLlib] Attention nets PyTorch support and cleanup (using traj. view API). (#12029)
2020-12-21 18:38:34 -08:00
..
__init__.py Pytorch AttentionNet (#9088) 2020-06-23 20:42:30 +02:00
convtranspose2d_stack.py [RLlib] ConvTranspose2D module (#11231) 2020-10-12 15:00:42 +02:00
gru_gate.py [RLlib] Attention nets PyTorch support and cleanup (using traj. view API). (#12029) 2020-12-21 18:38:34 -08:00
multi_head_attention.py [RLlib] Model Annotations to Torch Models (#9749) 2020-11-12 12:16:12 +01:00
noisy_layer.py [RLlib] Model Annotations to Torch Models (#9749) 2020-11-12 12:16:12 +01:00
relative_multi_head_attention.py [RLlib] Attention nets PyTorch support and cleanup (using traj. view API). (#12029) 2020-12-21 18:38:34 -08:00
skip_connection.py [RLlib] Attention Nets: tf (#12753) 2020-12-20 20:22:32 -05:00