Eric Liang 590943a499 [rllib] Type annotations for model classes (#9646)

2020-07-24 12:01:46 -07:00

..

modules

Pytorch AttentionNet (#9088)

2020-06-23 20:42:30 +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

Pytorch AttentionNet (#9088)

2020-06-23 20:42:30 +02:00

recurrent_net.py

This PR fixes the currently broken lstm_use_prev_action_reward flag for default lstm models (model.use_lstm=True). (#8970)

2020-06-27 20:50:01 +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