Merge pull request tensorflow#4822 from buckman-google/master

lukaszkaiser · web-flow · commit 76d3d72e3854 · 2018-07-20T12:35:29.000-07:00
Fixed minor bug in STEVE which sometimes prevented resuming experiments.
diff --git a/research/steve/learner.py b/research/steve/learner.py
@@ -155,9 +155,10 @@ def _resume_from_checkpoint(self):
         self.resume_from_checkpoint(epoch)
 
     def _log(self):
-        logstring = "(%3.2f sec) h%-8.2f e%-8d s%-8d f%-8d\t" % (time.time() - self._log_time, self.hours, self.epoch, self.update_i + 1, self.total_frames) + ', '.join(["%8f" % x for x in (self.running_total / self.denom).tolist()])
-        print("%s\t%s" % (self.learner_name, logstring))
-        with open(self.log_path, "a") as f: f.write(logstring + "\n")
+        if self.denom > 0:
+            logstring = "(%3.2f sec) h%-8.2f e%-8d s%-8d f%-8d\t" % (time.time() - self._log_time, self.hours, self.epoch, self.update_i + 1, self.total_frames) + ', '.join(["%8f" % x for x in (self.running_total / self.denom).tolist()])
+            print("%s\t%s" % (self.learner_name, logstring))
+            with open(self.log_path, "a") as f: f.write(logstring + "\n")
         self._reset_inspections()
 
     def _reset_inspections(self):
