stable-baselines3/tests
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
..
__init__.py Init: TD3 2019-09-05 17:29:41 +02:00
test_callbacks.py Add StopTrainingOnMaxEpisodes to callback collection (#147) 2020-08-28 11:36:33 +02:00
test_cnn.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_custom_policy.py Update black version + update docker image (#151) 2020-08-27 23:02:59 +02:00
test_deterministic.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_distributions.py Update black version + update docker image (#151) 2020-08-27 23:02:59 +02:00
test_envs.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_identity.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_logger.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_monitor.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_predict.py Allow to set a device when loading a model (#154) 2020-09-20 19:13:18 +02:00
test_run.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_save_load.py Get/set parameters and review of saving and loading (#138) 2020-09-24 14:28:27 +02:00
test_sde.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_spaces.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_tensorboard.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_utils.py Optimized polyak updates (#106) 2020-07-17 15:53:28 +02:00
test_vec_check_nan.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_vec_envs.py Update black version + update docker image (#151) 2020-08-27 23:02:59 +02:00
test_vec_normalize.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00