Commit 4a2a3122 authored by joseph_suarez4's avatar joseph_suarez4

Load model fix

parent c1f0a680
......@@ -368,7 +368,7 @@ class SanePPOTrainer(ppo.PPOTrainer):
'''Restore model from path'''
self.training_logs = np.load(
self.envConfig.PATH_TRAINING_DATA,
allow_pickle=True).item()
allow_pickle=True).item()['logs']
path = os.path.join(
self.envConfig.PATH_MODEL,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment