stable-baselines3/stable_baselines3
Alex Pasquali b702884c23
Removed shared layers in mlp_extractor (#1292)
* Modified actor-critic policies & MlpExtractor class

ActorCriticPolicy:
  - changed type hint of net_arch param: now it's a dict
  - removed check that if features extractor is not shared: no shared layers are allowed in the mlp_extractor regardless of the features extractor
ActorCriticCnnPolicy:
  - changed type hint of net_arch param: now it's a dict
MultiInputActorcriticPolicy:
  - changed type hint of net_arch param: now it's a dict
MlpExtractor:
  - changed type hint of net_arch param: now it's a dict
  - adapted networks creation
  - adapted methods: forward, forward_actor & forward_critic

* Removed shared layers in mlp_extractor

* Updated docs and changelog + reformat

* Updated custom policy tests

* Removed test on deprecation warning for share layers in mlp_extractor

Now shared layers are removed

* Update version

* Update RL Zoo doc

* Fix linter warnings

* Add ruff to Makefile (experimental)

* Add backward compat code and minor updates

* Update tests

* Add backward compatibility

* Fix test

* Improve compat code

Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
2023-01-23 14:55:19 +01:00
..
a2c Exposed modules in __init__.py with __all__ (#1195) 2022-11-29 23:33:46 +01:00
common Removed shared layers in mlp_extractor (#1292) 2023-01-23 14:55:19 +01:00
ddpg Exposed modules in __init__.py with __all__ (#1195) 2022-11-29 23:33:46 +01:00
dqn Standardize the use of from gym import spaces (#1240) 2023-01-02 14:51:11 +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 Standardize the use of from gym import spaces (#1240) 2023-01-02 14:51:11 +01:00
td3 Standardize the use of from gym import spaces (#1240) 2023-01-02 14:51:11 +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 Removed shared layers in mlp_extractor (#1292) 2023-01-23 14:55:19 +01:00