stable-baselines3/stable_baselines3/common
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
..
envs Downgrade sphinx-autodoc-typehints (#1291) 2023-01-23 10:56:45 +01:00
sb2_compat Upgrade code to Python 3.7+ syntax using pyupgrade (#887) 2022-04-25 13:01:38 +03:00
vec_env Standardize the use of from gym import spaces (#1240) 2023-01-02 14:51:11 +01:00
__init__.py Update docs (custom policy, type hints) (#167) 2020-09-29 20:41:14 +03:00
atari_wrappers.py Fix stable_baselines3/common/atari_wrappers.py type hints (#1216) 2022-12-18 16:13:44 +01:00
base_class.py Removed shared layers in mlp_extractor (#1292) 2023-01-23 14:55:19 +01:00
buffers.py Removed shared layers in mlp_extractor (#1292) 2023-01-23 14:55:19 +01:00
callbacks.py Fix test_vec_normalize.py, test_tensorboard.py and common/monitor.py type hint (#1194) 2023-01-13 18:28:22 +01:00
distributions.py Standardize the use of from gym import spaces (#1240) 2023-01-02 14:51:11 +01:00
env_checker.py Standardize the use of from gym import spaces (#1240) 2023-01-02 14:51:11 +01:00
env_util.py Fix stable_baselines3/common/env_util.py type hint (#1192) 2022-11-29 15:36:55 +01:00
evaluation.py Update evaluate_policy type annotation to support policies as well as RL algorithms (#1146) 2022-11-03 15:36:19 +01:00
logger.py Fix test_vec_normalize.py, test_tensorboard.py and common/monitor.py type hint (#1194) 2023-01-13 18:28:22 +01:00
monitor.py Fix test_vec_normalize.py, test_tensorboard.py and common/monitor.py type hint (#1194) 2023-01-13 18:28:22 +01:00
noise.py Drop gym.GoalEnv and other minor changes initally from #780 (#1184) 2022-11-28 18:22:31 +01:00
off_policy_algorithm.py Standardize the use of from gym import spaces (#1240) 2023-01-02 14:51:11 +01:00
on_policy_algorithm.py Standardize the use of from gym import spaces (#1240) 2023-01-02 14:51:11 +01:00
policies.py Removed shared layers in mlp_extractor (#1292) 2023-01-23 14:55:19 +01:00
preprocessing.py Fix support of image like normalized inputs (#1214) 2022-12-20 13:18:28 +01:00
results_plotter.py Fix default arguments + add bugbear (#363) 2021-03-25 11:35:21 +02:00
running_mean_std.py Fix test_vec_normalize.py, test_tensorboard.py and common/monitor.py type hint (#1194) 2023-01-13 18:28:22 +01:00
save_util.py Allow model trained with python3.7 to be loaded with python3.8+ without the custom_objects workaround (#1123) 2022-10-17 17:33:47 +02:00
torch_layers.py Removed shared layers in mlp_extractor (#1292) 2023-01-23 14:55:19 +01:00
type_aliases.py Fix stable_baselines3/common/type_aliases.py type hint (#1189) 2022-11-29 12:26:16 +01:00
utils.py Fix test_vec_normalize.py, test_tensorboard.py and common/monitor.py type hint (#1194) 2023-01-13 18:28:22 +01:00