ray/rllib/models/torch
Sven Mika 166bb5d690
[RLlib] IMPALA PyTorch (#8287)
This PR adds an IMPALA PyTorch implementation.

- adds compilation tests for LSTM and w/o LSTM.
- adds learning test for CartPole.
2020-05-03 13:44:25 +02:00
..
__init__.py [RLlib] Working/learning example: PPO + torch + LSTM. (#7797) 2020-03-31 22:00:28 -07:00
fcnet.py [RLlib] PyTorch version of ARS (Augmented Random Search). (#8106) 2020-04-21 09:47:52 +02:00
misc.py [RLlib] SAC Torch (incl. Atari learning) (#7984) 2020-04-15 13:25:16 +02:00
recurrent_torch_model.py [RLlib] Deprecate all Model(v1) usage. (#8146) 2020-04-29 12:12:59 +02:00
torch_action_dist.py [RLlib] IMPALA PyTorch (#8287) 2020-05-03 13:44:25 +02:00
torch_modelv2.py [RLlib] Deprecate all Model(v1) usage. (#8146) 2020-04-29 12:12:59 +02:00
visionnet.py [RLlib] SAC Torch (incl. Atari learning) (#7984) 2020-04-15 13:25:16 +02:00