ray/rllib/models/torch
2020-08-06 10:54:20 +02:00
..
modules [RLlib] Implement DQN PyTorch distributional head. (#9589) 2020-07-25 09:29:24 +02:00
__init__.py [RLlib] Attention Net integration into ModelV2 and learning RL example. (#8371) 2020-05-18 17:26:40 +02:00
attention_net.py Pytorch AttentionNet (#9088) 2020-06-23 20:42:30 +02:00
fcnet.py [RLlib] Make sure torch and tf behave the same wrt conv2d nets. (#8785) 2020-06-20 00:05:19 +02:00
misc.py [RLlib] Implement DQN PyTorch distributional head. (#9589) 2020-07-25 09:29:24 +02:00
recurrent_net.py [RLlib] Trajectory view API - 02 actual API scaffold (#9753) 2020-08-06 10:54:20 +02:00
torch_action_dist.py [rllib] Type annotations for model classes (#9646) 2020-07-24 12:01:46 -07:00
torch_modelv2.py [rllib] Type annotations for model classes (#9646) 2020-07-24 12:01:46 -07:00
visionnet.py [RLlib] Make sure torch and tf behave the same wrt conv2d nets. (#8785) 2020-06-20 00:05:19 +02:00