checkpoints bug fix

This commit is contained in:
itaicaspi-intel committed 2018-05-26 17:49:13 +03:00
1 parent 462c6e314b
commit 7725dabc86
1 file changed
+1 -1
+1 -1
View File
@@ -550,7 +550,7 @@ class Agent(object):
if current_snapshot_period > model_snapshots_periods_passed:
model_snapshots_periods_passed = current_snapshot_period
self.save_model(model_snapshots_periods_passed)
if self.running_observation_stats is not None:
if hasattr(self, 'running_observation_state') and self.running_observation_stats is not None:
to_pickle(self.running_observation_stats,
os.path.join(self.tp.save_model_dir,
"running_stats.p".format(model_snapshots_periods_passed)))