Support update specific steps (#437)

215a3244 · Yang Li · GitHub · 82e5c8e5 · 215a3244
Unverified Commit 215a3244 authored Jul 20, 2020 by Yang Li Committed by GitHub Jul 20, 2020
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

mmcv/utils/progressbar.py mmcv/utils/progressbar.py +3 -2

No files found.
--- a/mmcv/utils/progressbar.py
+++ b/mmcv/utils/progressbar.py
@@ -32,8 +32,9 @@ class ProgressBar:
        self.file.flush()
        self.timer = Timer()
-    def update(self):
+    def update(self, num_tasks=1):
-        self.completed += 1
+        assert num_tasks > 0
+        self.completed += num_tasks
        elapsed = self.timer.since_start()
        if elapsed > 0:
            fps = self.completed / elapsed