saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-05-17 21:20:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
stable-baselines3/tests
History
Dormann, Noah 1f0dd60b97 Fix saving on GPU - Loading on CPU (#45) 
* removed policy from save, changed th.loads to map to device

* found hack: catch pickle exception and trying th.load with mapping instead, otherwise raise exception with more information -> loading cuda on cpu raises exception -> leads to th.load with map being called

* deleted todo

* updated changelog

* start of saving refactor

* first working c

* all tests pass, save refactored

* - backwards compatibilty not always
- make pytest all passing
- make typing all passing

* Fixes and simplify the save method

* Remove unused param

* Fix backward compat

* Fix docstring
2020-01-31 13:06:55 +01:00
..
__init__.py Init: TD3 2019-09-05 17:29:41 +02:00
test_custom_policy.py Clean up code + bug fixes 2020-01-20 11:17:55 +01:00
test_distributions.py Clean up code + bug fixes 2020-01-20 11:17:55 +01:00
test_logger.py Improve logger testing + add readers 2020-01-20 11:58:00 +01:00
test_run.py Merge branch 'master' into feat/sde-features 2019-12-05 16:33:41 +01:00
test_save_load.py Fix saving on GPU - Loading on CPU (#45) 2020-01-31 13:06:55 +01:00
test_sde.py Clean up code + bug fixes 2020-01-20 11:17:55 +01:00
test_vec_envs.py Clean up code + bug fixes 2020-01-20 11:17:55 +01:00
test_vec_normalize.py Update VecNormalize (pickling) and improve tests 2020-01-20 11:58:16 +01:00