saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-04 04:07:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
stable-baselines3/tests
History
Anand Balakrishnan 59af0c1b01
CheckpointCallback can now save replay buffer and VecNormalize (#1030) 
* CheckpointCallback now saves replay buffer (if present)

* VecNormalize stats are saved at checkpoints

* Make checkpointing replay buffer and VecNormalize opt-in

* Edit changelog

* Add documentation for new parameters

* Update docs/misc/changelog.rst

* Add documentation for new parameters

* Implement suggested edits

* Reformat code

* Fix git conflict

* Add .pkl suffix to VecNormalize checkpoints

* Add tests for new CheckpointCallback params

* Merge CheckpointCallback tests

* Update test and add helper for checkpoint path

Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
2022-08-25 10:57:51 +02:00
..
__init__.py
test_buffers.py Support for device=auto buffers and set it as default value (#1009) 2022-08-16 17:54:55 +02:00
test_callbacks.py CheckpointCallback can now save replay buffer and VecNormalize (#1030) 2022-08-25 10:57:51 +02:00
test_cnn.py Add skip option for VecTransposeImage and bug fix in frame stack (#700) 2021-12-23 17:12:49 +02:00
test_custom_policy.py Gym fixes - Follow up from #705 (#734) 2022-02-04 15:13:57 -08:00
test_deterministic.py Gym fixes - Follow up from #705 (#734) 2022-02-04 15:13:57 -08:00
test_dict_env.py Multiprocessing support for off policy algorithms (#439) 2021-12-01 22:30:09 +01:00
test_distributions.py Release v1.6.0 (#958) 2022-07-12 22:50:23 +02:00
test_env_checker.py Gym fixes - Follow up from #705 (#734) 2022-02-04 15:13:57 -08:00
test_envs.py Update doc and add check for unbounded action space (#918) 2022-05-25 16:24:21 +02:00
test_gae.py Upgrade code to Python 3.7+ syntax using pyupgrade (#887) 2022-04-25 13:01:38 +03:00
test_her.py Upgrade code to Python 3.7+ syntax using pyupgrade (#887) 2022-04-25 13:01:38 +03:00
test_identity.py
test_logger.py Support hparams logging to tensorboard (#984) 2022-08-22 22:06:54 +02:00
test_monitor.py Upgrade code to Python 3.7+ syntax using pyupgrade (#887) 2022-04-25 13:01:38 +03:00
test_predict.py Fix returned type in predict (#964) 2022-07-18 11:22:19 +02:00
test_run.py Use higher resolution time_ns() and avoid division by zero (#979) 2022-07-25 23:02:53 +02:00
test_save_load.py Prohibit simultaneous use of optimize_memory_usage and handle_timeout_termination (#948) 2022-07-04 15:08:54 +02:00
test_sde.py Gym fixes - Follow up from #705 (#734) 2022-02-04 15:13:57 -08:00
test_spaces.py Handling multi-dimensional action spaces (#971) 2022-08-06 14:19:20 +02:00
test_tensorboard.py Support hparams logging to tensorboard (#984) 2022-08-22 22:06:54 +02:00
test_train_eval_mode.py Include running_mean and running_val when updating target networks (#1004) 2022-08-23 10:20:43 +02:00
test_utils.py Include running_mean and running_val when updating target networks (#1004) 2022-08-23 10:20:43 +02:00
test_vec_check_nan.py Upgrade code to Python 3.7+ syntax using pyupgrade (#887) 2022-04-25 13:01:38 +03:00
test_vec_envs.py Fixed unchecked None value in SubprocVecEnv (#808) 2022-04-12 16:05:40 +02:00
test_vec_extract_dict_obs.py
test_vec_monitor.py Upgrade code to Python 3.7+ syntax using pyupgrade (#887) 2022-04-25 13:01:38 +03:00
test_vec_normalize.py Fix synchronization bug with EvalCallback (#907) 2022-05-08 21:54:34 +03:00