stable-baselines3/stable_baselines3
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
..
a2c Upgrade code to Python 3.7+ syntax using pyupgrade (#887) 2022-04-25 13:01:38 +03:00
common CheckpointCallback can now save replay buffer and VecNormalize (#1030) 2022-08-25 10:57:51 +02:00
ddpg Upgrade code to Python 3.7+ syntax using pyupgrade (#887) 2022-04-25 13:01:38 +03:00
dqn Include running_mean and running_val when updating target networks (#1004) 2022-08-23 10:20:43 +02:00
her Support for device=auto buffers and set it as default value (#1009) 2022-08-16 17:54:55 +02:00
ppo Upgrade code to Python 3.7+ syntax using pyupgrade (#887) 2022-04-25 13:01:38 +03:00
sac Include running_mean and running_val when updating target networks (#1004) 2022-08-23 10:20:43 +02:00
td3 Include running_mean and running_val when updating target networks (#1004) 2022-08-23 10:20:43 +02:00
__init__.py Upgrade code to Python 3.7+ syntax using pyupgrade (#887) 2022-04-25 13:01:38 +03:00
py.typed Rename to stable-baselines3 2020-05-05 15:02:35 +02:00
version.txt Include running_mean and running_val when updating target networks (#1004) 2022-08-23 10:20:43 +02:00