Sven Mika 56878221ed [RLlib] Redo: Make TFModelV2 fully modular like TorchModelV2 (soft-deprecate register_variables, unify var names wrt torch). (#13363)

2021-01-14 14:44:33 +01:00

..

modules

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

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

__init__.py

[RLlib] Deprecate old classes, methods, functions, config keys (in prep for RLlib 1.0). (#10544)

2020-09-06 10:58:00 +02:00

attention_net.py

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

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

fcnet.py

[RLlib] Add a minimal JAX ModelV2 (FCNet) to RLlib. (#12502)

2020-12-03 15:51:30 +01:00

misc.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

recurrent_net.py

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

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

torch_action_dist.py

[RLlib] SquashedGaussians should throw error when entropy or kl are called. (#13126)

2021-01-07 15:07:35 +01:00

torch_modelv2.py

[RLlib] Redo: Make TFModelV2 fully modular like TorchModelV2 (soft-deprecate register_variables, unify var names wrt torch). (#13363)

2021-01-14 14:44:33 +01:00

visionnet.py

[RLlib] Trajectory View API: Atari framestacking. (#13315)

2021-01-13 08:53:34 +01:00