stable-baselines3/torchy_baselines
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
..
a2c Add sample freq for SDE 2019-12-17 11:47:21 +01:00
cem_rl Fix typing errors and typos 2020-01-22 17:17:12 +01:00
common Fix saving on GPU - Loading on CPU (#45) 2020-01-31 13:06:55 +01:00
ppo Fix saving on GPU - Loading on CPU (#45) 2020-01-31 13:06:55 +01:00
sac Fix saving on GPU - Loading on CPU (#45) 2020-01-31 13:06:55 +01:00
td3 Fix saving on GPU - Loading on CPU (#45) 2020-01-31 13:06:55 +01:00
__init__.py Build doc 2020-01-20 16:19:35 +01:00
py.typed Revert all changes for python 2 2020-01-22 16:18:27 +01:00