fix ZeroDivisionError in progressbar (#162) (#163)

b34eef6e · Kevin · Kai Chen · 0d00f1a6 · b34eef6e
Commit b34eef6e authored Jan 02, 2020 by Kevin Committed by Kai Chen Jan 02, 2020
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

mmcv/utils/progressbar.py mmcv/utils/progressbar.py +4 -1

No files found.
--- a/mmcv/utils/progressbar.py
+++ b/mmcv/utils/progressbar.py
@@ -44,7 +44,10 @@ class ProgressBar(object):
    def update(self):
        self.completed += 1
        elapsed = self.timer.since_start()
+        if elapsed > 0:
            fps = self.completed / elapsed
+        else:
+            fps = float('inf')
        if self.task_num > 0:
            percentage = self.completed / float(self.task_num)
            eta = int(elapsed * (1 - percentage) / percentage + 0.5)