stable-baselines3/tests
Oleksii Kachaiev 0c17fedfac
Adjust FPS calculation to accommodate for reset_num_timesteps=False (#636)
* Store number of timesteps at the beginning of each learn cycle

* Update changelog

* Set default _num_timesteps_at_start in the contructor

* Test case for FPS logger

* Adjust test to cover both on-policy and off-policy algorithms

* Fix formatting

* Update test and add comment

* Fix test

Co-authored-by: Oleksii Kachaiev <okachaiev@riotgames.com>
Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
2021-10-31 18:19:03 +01:00
..
__init__.py Init: TD3 2019-09-05 17:29:41 +02:00
test_callbacks.py Doc fix and improve error messages (#598) 2021-10-08 18:08:31 +02:00
test_cnn.py Documentation update (#450) 2021-05-23 13:13:11 +02:00
test_custom_policy.py Fix discrete obs support (#296) 2021-01-21 02:42:33 +02:00
test_deterministic.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_dict_env.py Introduce norm_obs_keys param for VecNormalize environment wrapper (#631) 2021-10-28 19:18:39 +02:00
test_distributions.py Add method predict_values for ActorCriticPolicy (#569) 2021-09-15 14:03:04 +02:00
test_env_checker.py add check to ensure action space is non-dict non-tuple for env_checker nan check (#192) 2020-10-19 00:23:51 +03:00
test_envs.py Dictionary Observations (#243) 2021-05-11 12:29:30 +02:00
test_gae.py Add test for GAE + rename RolloutBuffer.dones for clarification (#375) 2021-04-16 15:52:55 +02:00
test_her.py Dictionary Observations (#243) 2021-05-11 12:29:30 +02:00
test_identity.py Dictionary Observations (#243) 2021-05-11 12:29:30 +02:00
test_logger.py Adjust FPS calculation to accommodate for reset_num_timesteps=False (#636) 2021-10-31 18:19:03 +01:00
test_monitor.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_predict.py Fixes in is_vectorized_observation (#587) 2021-09-28 21:57:49 +02:00
test_run.py Fix train_freq at load time (#332) 2021-02-27 19:53:13 +01:00
test_save_load.py Fix set_env to keep the number of timesteps (#615) 2021-10-23 16:36:40 +02:00
test_sde.py Remove sde_net_arch + Simplify policy (#584) 2021-09-28 22:32:54 +03:00
test_spaces.py Fix discrete obs support (#296) 2021-01-21 02:42:33 +02:00
test_tensorboard.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_train_eval_mode.py Avoid putting target networks into training mode (#553) 2021-08-30 17:42:41 +02:00
test_utils.py System info helper (#613) 2021-10-18 10:43:56 +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 Use Monitor episode reward/length for evaluate_policy (#220) 2020-11-16 11:52:28 +01: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 Support for VecMonitor for gym3-style environments (#311) 2021-04-13 18:09:31 +02:00
test_vec_normalize.py Introduce norm_obs_keys param for VecNormalize environment wrapper (#631) 2021-10-28 19:18:39 +02:00