1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +01:00

checkpoints bug fix

This commit is contained in:
itaicaspi-intel
2018-05-26 17:49:13 +03:00
parent 462c6e314b
commit 7725dabc86

View File

@@ -550,7 +550,7 @@ class Agent(object):
if current_snapshot_period > model_snapshots_periods_passed: if current_snapshot_period > model_snapshots_periods_passed:
model_snapshots_periods_passed = current_snapshot_period model_snapshots_periods_passed = current_snapshot_period
self.save_model(model_snapshots_periods_passed) 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, to_pickle(self.running_observation_stats,
os.path.join(self.tp.save_model_dir, os.path.join(self.tp.save_model_dir,
"running_stats.p".format(model_snapshots_periods_passed))) "running_stats.p".format(model_snapshots_periods_passed)))