stable-baselines3/torchy_baselines/ppo
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 Add doc 2019-09-26 11:46:40 +02:00
policies.py Fix typing errors and typos 2020-01-22 17:17:12 +01:00
ppo.py Fix saving on GPU - Loading on CPU (#45) 2020-01-31 13:06:55 +01:00