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