saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-30 20:18:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Hotfix: properly delete device in policy_kwargs

Browse source
This commit is contained in:
Antonin RAFFIN 2020-09-25 15:34:53 +02:00
parent 44a723eecb
commit 8b16324ba7
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
stable_baselines3/common/base_class.py
View file

@ -562,7 +562,7 @@ class BaseAlgorithm(ABC):
# Remove stored device information and replace with ours
if "policy_kwargs" in data:
if "device" in "policy_kwargs":
if "device" in data["policy_kwargs"]:
del data["policy_kwargs"]["device"]
if "policy_kwargs" in kwargs and kwargs["policy_kwargs"] != data["policy_kwargs"]: