stable-baselines3/stable_baselines3/common
Anssi 9855486488
Get/set parameters and review of saving and loading (#138)
* Update comments and docstrings

* Rename get_torch_variables to private and update docs

* Clarify documentation on data, params and tensors

* Make excluded_save_params private and update docs

* Update get_torch_variable_names to get_torch_save_params for description

* Simplify saving code and update docs on params vs tensors

* Rename saved item tensors to pytorch_variables for clarity

* Reformat

* Fix a typo

* Add get/set_parameters, update tests accordingly

* Use f-strings for formatting

* Fix load docstring

* Reorganize functions in BaseClass

* Update changelog

* Add library version to the stored models

* Actually run isort this time

* Fix flake8 complaints and also fix testing code

* Fix isort

* ...and black

* Fix set_random_seed

Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
Co-authored-by: Antonin Raffin <antonin.raffin@dlr.de>
2020-09-24 14:28:27 +02:00
..
sb2_compat Match performance with stable-baselines (discrete case) (#110) 2020-08-03 22:22:51 +02:00
vec_env Fix double reset and improve typing coverage (#136) 2020-08-05 13:12:02 +03:00
__init__.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
atari_wrappers.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
base_class.py Get/set parameters and review of saving and loading (#138) 2020-09-24 14:28:27 +02:00
bit_flipping_env.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
buffers.py Update black version + update docker image (#151) 2020-08-27 23:02:59 +02:00
callbacks.py Fix f-string in max episodes callback (#152) 2020-08-29 20:04:19 +02:00
cmd_util.py Fix type annotation in make_vec_env (#162) 2020-09-23 10:34:35 +02:00
distributions.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
env_checker.py Fix double reset and improve typing coverage (#136) 2020-08-05 13:12:02 +03:00
evaluation.py Fix double reset and improve typing coverage (#136) 2020-08-05 13:12:02 +03:00
identity_env.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
logger.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
monitor.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
noise.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
off_policy_algorithm.py Add StopTrainingOnMaxEpisodes to callback collection (#147) 2020-08-28 11:36:33 +02:00
on_policy_algorithm.py Get/set parameters and review of saving and loading (#138) 2020-09-24 14:28:27 +02:00
policies.py Update black version + update docker image (#151) 2020-08-27 23:02:59 +02:00
preprocessing.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
results_plotter.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
running_mean_std.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
save_util.py Get/set parameters and review of saving and loading (#138) 2020-09-24 14:28:27 +02:00
torch_layers.py Match performance with stable-baselines (discrete case) (#110) 2020-08-03 22:22:51 +02:00
type_aliases.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
utils.py Allow to set a device when loading a model (#154) 2020-09-20 19:13:18 +02:00