Sven MikaandGitHub 9eba1871bb [RLlib] Support easy use_attention=True flag for using the GTrXL model. (#11698)

2021-01-01 14:06:23 -05:00

..

__init__.py

[RLlib] Attention Nets: tf (#12753)

2020-12-20 20:22:32 -05:00

gru_gate.py

[RLlib] Model Annotations: Tensorflow (#11964)

2020-11-12 12:18:50 +01:00

multi_head_attention.py

[RLlib] Model Annotations: Tensorflow (#11964)

2020-11-12 12:18:50 +01:00

noisy_layer.py

[RLlib] JAXPolicy prep PR #2 (move get_activation_fn (backward-compatibly), minor fixes and preparations). (#13091)

2020-12-30 22:30:52 -05:00

relative_multi_head_attention.py

[RLlib] Support easy use_attention=True flag for using the GTrXL model. (#11698)

2021-01-01 14:06:23 -05:00

skip_connection.py

[RLlib] Attention Nets: tf (#12753)

2020-12-20 20:22:32 -05:00