bugfix

ocean2811 · Jun 3, 2019 · a83f31d · a83f31d
1 parent cc78811
commit a83f31d
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/scripts/train.py b/scripts/train.py
@@ -208,10 +208,9 @@ def run_training_cycle(self, model, trainer):
 
         for iteration in range(0, self.args.iterations):
             logger.trace("Training iteration: %s", iteration)
-            snapshot_interval = self.args.save_interval * self.args.snapshot_interval
-            snapshot_iteration = bool(snapshot_interval != 0 and
-                                      iteration >= snapshot_interval and
-                                      iteration % snapshot_interval == 0)
+            snapshot_iteration = bool(self.args.snapshot_interval != 0 and
+                                      iteration >= self.args.snapshot_interval and
+                                      iteration % self.args.snapshot_interval == 0)
             save_iteration = iteration % self.args.save_interval == 0
             viewer = display_func if save_iteration or self.save_now else None
             timelapse = self.timelapse if save_iteration else None