gaspersic/baselines
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

comment about loading weights in ppo2

Browse Source
This commit is contained in:
Oleg Klimov
2018-02-05 12:25:05 -08:00
parent 2793971c10
commit d9b36601d9
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
baselines/ppo2/ppo2.py
View File

@@ -72,6 +72,7 @@ class Model(object):
for p, loaded_p in zip(params, loaded_params): for p, loaded_p in zip(params, loaded_params):
restores.append(p.assign(loaded_p)) restores.append(p.assign(loaded_p))
sess.run(restores) sess.run(restores)
# If you want to load weights, also save/load observation scaling inside VecNormalize
self.train = train self.train = train
self.train_model = train_model self.train_model = train_model