stable-baselines3/stable_baselines3
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
..
a2c Exposed modules in __init__.py with __all__ (#1195) 2022-11-29 23:33:46 +01:00
common Modified ActorCriticPolicy to support non-shared features extractor (#1148) 2022-12-20 15:12:05 +01:00
ddpg Exposed modules in __init__.py with __all__ (#1195) 2022-11-29 23:33:46 +01:00
dqn Modified ActorCriticPolicy to support non-shared features extractor (#1148) 2022-12-20 15:12:05 +01:00
her Exposed modules in __init__.py with __all__ (#1195) 2022-11-29 23:33:46 +01:00
ppo Exposed modules in __init__.py with __all__ (#1195) 2022-11-29 23:33:46 +01:00
sac Modified ActorCriticPolicy to support non-shared features extractor (#1148) 2022-12-20 15:12:05 +01:00
td3 Modified ActorCriticPolicy to support non-shared features extractor (#1148) 2022-12-20 15:12:05 +01:00
__init__.py Update version (#1220) 2022-12-19 13:53:00 +01:00
py.typed Rename to stable-baselines3 2020-05-05 15:02:35 +02:00
version.txt Modified ActorCriticPolicy to support non-shared features extractor (#1148) 2022-12-20 15:12:05 +01:00