stable-baselines3/tests
Alex Pasquali 2cfcec4f50
Modified ActorCriticPolicy to support non-shared features extractor (#1148)
* Modified ActorCriticPolicy to support non-shared features extractor

* Refactored features extraction with non-shared features extractor in ActorCriticPolicy and updated doc

Doc update: added 'warning' on custom policy docs that says that, if the features extractor is non-shared, it's not possible to have shared layers in the mlp_extractor

* Moved attrib share_features_extractor in class

* Updated custom policy doc for non-shared features extractor

* Updated changelog

* Made some if-statements more readable if policies.py

The if-statements are related to the shared/non-shared features extractor in ActorCritic policies

* Simplify implementation and add run test

* Keep order in module gain to keep previous results consistents

* Fix test

* Improved docstring in policies.py

Co-authored-by: Quentin Gallouédec <45557362+qgallouedec@users.noreply.github.com>

* Added some tests

* feature extractor -> features extractor

* Fix test

* Fix env_id in test

* Make features extractor parameter explicit

* Remove duplicate

Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
Co-authored-by: Quentin Gallouédec <45557362+qgallouedec@users.noreply.github.com>
Co-authored-by: Antonin Raffin <antonin.raffin@dlr.de>
2022-12-20 15:12:05 +01:00
..
__init__.py Init: TD3 2019-09-05 17:29:41 +02:00
test_buffers.py Support for device=auto buffers and set it as default value (#1009) 2022-08-16 17:54:55 +02:00
test_callbacks.py Drop gym.GoalEnv and other minor changes initally from #780 (#1184) 2022-11-28 18:22:31 +01:00
test_cnn.py Modified ActorCriticPolicy to support non-shared features extractor (#1148) 2022-12-20 15:12:05 +01:00
test_custom_policy.py Modified ActorCriticPolicy to support non-shared features extractor (#1148) 2022-12-20 15:12:05 +01:00
test_deterministic.py Gym fixes - Follow up from #705 (#734) 2022-02-04 15:13:57 -08:00
test_dict_env.py Fix support of image like normalized inputs (#1214) 2022-12-20 13:18:28 +01:00
test_distributions.py Fix tests/test_distributions.py type hint (#1186) 2022-11-29 11:27:59 +01:00
test_env_checker.py Gym fixes - Follow up from #705 (#734) 2022-02-04 15:13:57 -08:00
test_envs.py Drop gym.GoalEnv and other minor changes initally from #780 (#1184) 2022-11-28 18:22:31 +01:00
test_gae.py Upgrade code to Python 3.7+ syntax using pyupgrade (#887) 2022-04-25 13:01:38 +03:00
test_her.py Upgrade code to Python 3.7+ syntax using pyupgrade (#887) 2022-04-25 13:01:38 +03:00
test_identity.py Add timeout handling for on-policy algorithms (#658) 2021-11-16 17:19:16 +01:00
test_logger.py Fix duplicate key error in HumanOutputFormat (#1079) 2022-09-28 12:06:07 +02:00
test_monitor.py Automatically create missing directories of `filenames passed to ResultsWriter` (#1072) 2022-09-21 13:14:38 +02:00
test_predict.py Drop gym.GoalEnv and other minor changes initally from #780 (#1184) 2022-11-28 18:22:31 +01:00
test_preprocessing.py Add support for multidimensional spaces.MultiBinary observations (#1179) 2022-12-08 18:46:41 +01:00
test_run.py Remove deprecated features and attributes (#1104) 2022-10-11 10:55:16 +02:00
test_save_load.py Drop gym.GoalEnv and other minor changes initally from #780 (#1184) 2022-11-28 18:22:31 +01:00
test_sde.py Remove deprecated features and attributes (#1104) 2022-10-11 10:55:16 +02:00
test_spaces.py Add support for multidimensional spaces.MultiBinary observations (#1179) 2022-12-08 18:46:41 +01:00
test_tensorboard.py Support hparams logging to tensorboard (#984) 2022-08-22 22:06:54 +02:00
test_train_eval_mode.py Remove deprecated features and attributes (#1104) 2022-10-11 10:55:16 +02:00
test_utils.py Drop gym.GoalEnv and other minor changes initally from #780 (#1184) 2022-11-28 18:22:31 +01:00
test_vec_check_nan.py Upgrade code to Python 3.7+ syntax using pyupgrade (#887) 2022-04-25 13:01:38 +03:00
test_vec_envs.py Raise error when same env object instance is passed in vectorized environment (#1154) 2022-11-22 14:28:58 +01:00
test_vec_extract_dict_obs.py Support for VecMonitor for gym3-style environments (#311) 2021-04-13 18:09:31 +02:00
test_vec_monitor.py Drop gym.GoalEnv and other minor changes initally from #780 (#1184) 2022-11-28 18:22:31 +01:00
test_vec_normalize.py Drop gym.GoalEnv and other minor changes initally from #780 (#1184) 2022-11-28 18:22:31 +01:00