saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-03 03:59:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
stable-baselines3/tests
History
Honglu Fan 29a481a288
Include running_mean and running_val when updating target networks (#1004) 
* include `running_mean` and `running_val` when updating target networks in DQN, SAC, TD3.

* Update stable_baselines3/common/utils.py

Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>

* Precompute batch norm parameters in `_setup_model` and directly copy them in the target update.

* include `running_mean` and `running_val` when updating target networks in DQN, SAC, TD3.

* Update stable_baselines3/common/utils.py

Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>

* Precompute batch norm parameters in `_setup_model` and directly copy them in the target update.

* Fix `DictReplayBuffer.next_observations` type (#1013)

* Fix DictReplayBuffer.next_observations type

* Update changelog

Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>

* Fixed missing verbose parameter passing (#1011)

Co-authored-by: Quentin Gallouédec <45557362+qgallouedec@users.noreply.github.com>

* Support for `device=auto` buffers and set it as default value (#1009)

* Default device is "auto" for buffer + auto device support in BufferBaseClass

* Update docstring

* Update tests

* Unify tests

* Update changelog

* Fix tests on CUDA device

Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
Co-authored-by: Antonin Raffin <antonin.raffin@dlr.de>

* Precompute batch norm parameters in `_setup_model` and directly copy them in the target update.

* Update test

* Add comments and update tests

* Bump version

* Remove one extra space to conform code style.

* Update docstrings

Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
Co-authored-by: Quentin Gallouédec <45557362+qgallouedec@users.noreply.github.com>
Co-authored-by: Burak Demirbilek <BurakDmb@users.noreply.github.com>
Co-authored-by: Antonin Raffin <antonin.raffin@dlr.de>
2022-08-23 10:20:43 +02:00
..
__init__.py Init: TD3 2019-09-05 17:29:41 +02:00
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 Callback to early stop the training if there is no model improvement after consecutive evaluations (#741) 2022-02-25 11:56:47 +01: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 Add timeout handling for on-policy algorithms (#658) 2021-11-16 17:19:16 +01:00
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 Support for VecMonitor for gym3-style environments (#311) 2021-04-13 18:09:31 +02:00
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