stable-baselines3/torchy_baselines/common
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
..
vec_env Update docstring 2020-01-23 11:20:12 +01:00
__init__.py Init: TD3 2019-09-05 17:29:41 +02:00
base_class.py Fix saving on GPU - Loading on CPU (#45) 2020-01-31 13:06:55 +01:00
buffers.py Clean up code + bug fixes 2020-01-20 11:17:55 +01:00
distributions.py Convert format to f-strings 2020-01-22 16:39:25 +01:00
evaluation.py Convert format to f-strings 2020-01-22 16:39:25 +01:00
identity_env.py first save and load features 2019-11-12 17:03:57 +01:00
logger.py Convert format to f-strings 2020-01-22 16:39:25 +01:00
monitor.py Add default filename for monitor 2019-10-29 18:42:34 +01:00
noise.py Convert format to f-strings 2020-01-22 16:39:25 +01:00
policies.py Convert format to f-strings 2020-01-22 16:39:25 +01:00
running_mean_std.py Add stable-baselines VecEnvs 2019-09-20 15:18:25 +02:00
save_util.py Fix saving on GPU - Loading on CPU (#45) 2020-01-31 13:06:55 +01:00
utils.py Add learning rate schedule 2019-10-28 16:47:13 +01:00