stable-baselines3/stable_baselines3/common
Anssi 44f8218df0
Review of code (A2C, PPO and refactoring) (#35)
* Split torch module code into torch_layers file

* Updated reference to CNN

* Change 'CxWxH' to 'CxHxW', as per common notion

* Fix missing import in policies.py

* Move PPOPolicy to OnlineActorCriticPolicy

* Create OnPolicyRLModel from PPO, and make A2C and PPO inherit

* Update A2C optimizer comment

* Clean weight init scales for clarity

* Fix A2C log_interval default parameter

* Rename 'progress' to 'progress_remaining

* Rename 'Models' to 'Algorithms'

* Rename 'OnlineActorCriticPolicy' to 'ActorCriticPolicy'

* Move static functions out from BaseAlgorithm

* Move on/off_policy base algorithms to their own files

* Add  files for A2C/PPO

* Fix docs

* Fix pytype

* Update documentation on OnPolicyAlgorithm

* Add proper doctstring for on_policy rollout gathering

* Add bit clarification on the mlppolicy/cnnpolicy naming

* Move static function is_vectorized_policies to utils.py

* Checking docstrings, pep8 fixes

* Update changelog

* Clean changelog

* Remove policy warnings for sac/td3

* Add monitor_wrapper for OnPolicyAlgorithm. Clean tb logging variables. Add parameter keywords to OffPolicyAlgorithm super init

Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
2020-06-09 13:54:18 +02:00
..
vec_env Review of code (A2C, PPO and refactoring) (#35) 2020-06-09 13:54:18 +02:00
__init__.py More doc + sync VecEnvs + atari 2020-05-07 16:08:23 +02:00
atari_wrappers.py Update doc 2020-05-08 13:09:38 +02:00
base_class.py Review of code (A2C, PPO and refactoring) (#35) 2020-06-09 13:54:18 +02:00
bit_flipping_env.py Sync with Stable-Baselines 2020-05-05 16:28:38 +02:00
buffers.py Support for MultiBinary / MultiDiscrete spaces (#13) 2020-05-18 14:42:13 +02:00
callbacks.py Review of code (A2C, PPO and refactoring) (#35) 2020-06-09 13:54:18 +02:00
cmd_util.py Add flake8 linter and Github CI (#19) 2020-05-12 17:55:01 +02:00
distributions.py Support for MultiBinary / MultiDiscrete spaces (#13) 2020-05-18 14:42:13 +02:00
env_checker.py Build the doc 2020-05-07 17:35:29 +02:00
evaluation.py Review of code (A2C, PPO and refactoring) (#35) 2020-06-09 13:54:18 +02:00
identity_env.py Add flake8 linter and Github CI (#19) 2020-05-12 17:55:01 +02:00
logger.py Tensorboard integration (#30) 2020-06-01 11:55:44 +02:00
monitor.py Update doc 2020-05-08 13:09:38 +02:00
noise.py Implemented Vectorized Action Noise (#34) 2020-05-27 09:53:01 +02:00
off_policy_algorithm.py Review of code (A2C, PPO and refactoring) (#35) 2020-06-09 13:54:18 +02:00
on_policy_algorithm.py Review of code (A2C, PPO and refactoring) (#35) 2020-06-09 13:54:18 +02:00
policies.py Review of code (A2C, PPO and refactoring) (#35) 2020-06-09 13:54:18 +02:00
preprocessing.py Support for MultiBinary / MultiDiscrete spaces (#13) 2020-05-18 14:42:13 +02:00
results_plotter.py Rename to stable-baselines3 2020-05-05 15:02:35 +02:00
running_mean_std.py Rename to stable-baselines3 2020-05-05 15:02:35 +02:00
save_util.py Review of code (A2C, PPO and refactoring) (#35) 2020-06-09 13:54:18 +02:00
torch_layers.py Review of code (A2C, PPO and refactoring) (#35) 2020-06-09 13:54:18 +02:00
type_aliases.py Rename to stable-baselines3 2020-05-05 15:02:35 +02:00
utils.py Review of code (A2C, PPO and refactoring) (#35) 2020-06-09 13:54:18 +02:00