saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-11 17:48:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

_setup_model() is now called when model is loaded

Browse source
This commit is contained in:
Noah Dormann 2019-11-28 15:33:52 +01:00
parent 9ff59eaf3d
commit 751ccf85e7
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
torchy_baselines/common/base_class.py
View file

@ -287,13 +287,13 @@ class BaseRLModel(object):
"Stored kwargs: {}, specified kwargs: {}".format(data['policy_kwargs'],
kwargs['policy_kwargs']))
model = cls(policy=data["policy"], env=data["env"], _init_setup_model=False)
model = cls(policy=data["policy_class"], env=data["env"], _init_setup_model=True)
model.__dict__.update(data)
model.__dict__.update(kwargs)
model.set_env(env)
model.load_parameters(params, opt_params)
# resetup modul after load
#model._setup_model()
# model._setup_model()
return model
@staticmethod