comment about loading weights in ppo2

This commit is contained in:
Oleg Klimov
2018-02-05 12:25:05 -08:00
parent 2793971c10
commit d9b36601d9

View File

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