stable-baselines3/tests
Antonin RAFFIN 2599f04940
Add custom arch for off-policy actor/critic networks (#182)
* Add custom arch for off-policy actor/critic networks

* Fix type hints

* Address comments

* Make sure number of updated parameters match in polyak

* Add zip_strict for strict-length zipping

* Fix building docs

* Add test for zip strict

* Faster tests

Co-authored-by: Anssi "Miffyli" Kanervisto <kaneran21@hotmail.com>
2020-10-13 12:01:33 +02:00
..
__init__.py Init: TD3 2019-09-05 17:29:41 +02:00
test_callbacks.py Add StopTrainingOnMaxEpisodes to callback collection (#147) 2020-08-28 11:36:33 +02:00
test_cnn.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_custom_policy.py Add custom arch for off-policy actor/critic networks (#182) 2020-10-13 12:01:33 +02:00
test_deterministic.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_distributions.py Update black version + update docker image (#151) 2020-08-27 23:02:59 +02:00
test_envs.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_identity.py Fix off-by-one GAE computation (#185) 2020-10-13 00:10:54 +03:00
test_logger.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_monitor.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_predict.py Allow to set a device when loading a model (#154) 2020-09-20 19:13:18 +02:00
test_run.py Add custom arch for off-policy actor/critic networks (#182) 2020-10-13 12:01:33 +02:00
test_save_load.py Add custom arch for off-policy actor/critic networks (#182) 2020-10-13 12:01:33 +02:00
test_sde.py Add custom arch for off-policy actor/critic networks (#182) 2020-10-13 12:01:33 +02:00
test_spaces.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_tensorboard.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_utils.py Add custom arch for off-policy actor/critic networks (#182) 2020-10-13 12:01:33 +02:00
test_vec_check_nan.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_vec_envs.py Update black version + update docker image (#151) 2020-08-27 23:02:59 +02:00
test_vec_normalize.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00