ray/rllib/models/torch/modules
2021-11-03 10:00:46 +01:00
..
__init__.py Pytorch AttentionNet (#9088) 2020-06-23 20:42:30 +02:00
convtranspose2d_stack.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
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] Fix deprecated warning for torch_ops.py (soft-replaced by torch_utils.py). (#19982) 2021-11-03 10:00:46 +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] Fix deprecated warning for torch_ops.py (soft-replaced by torch_utils.py). (#19982) 2021-11-03 10:00:46 +01:00
skip_connection.py [RLlib] Attention Nets: tf (#12753) 2020-12-20 20:22:32 -05:00