mirror of
https://github.com/saymrwulf/stable-baselines3.git
synced 2026-05-17 21:20:11 +00:00
f5104a5efc
* Added a 'device' keyword argument to BaseAlgorithm.load(). Edited the save and load test to also test the load method with all possible devices. Added the changes to the changelog * improved the load test to ensure that the model loads to the correct device. * improved the test: now the correctness is improved. If the get_device policy would change, it wouldn't break the test. * Update tests/test_save_load.py @araffin's suggestion during the PR process Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org> * Update tests/test_save_load.py Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org> * Bug fixes: when comparing devices, comparing only device type since get_device() doesn't provide device index. Now the code loads all of the model parameters from the saved state dict straight into the required device. (fixed load_from_zip_file). * PR fixes: bug fix - a non-related test failed when running on GPU. updated the assertion to consider only types of devices. Also corrected a related bug in 'get_device()' method. * Update changelog.rst Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org> |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| test_callbacks.py | ||
| test_cnn.py | ||
| test_custom_policy.py | ||
| test_deterministic.py | ||
| test_distributions.py | ||
| test_envs.py | ||
| test_identity.py | ||
| test_logger.py | ||
| test_monitor.py | ||
| test_predict.py | ||
| test_run.py | ||
| test_save_load.py | ||
| test_sde.py | ||
| test_spaces.py | ||
| test_tensorboard.py | ||
| test_utils.py | ||
| test_vec_check_nan.py | ||
| test_vec_envs.py | ||
| test_vec_normalize.py | ||