[Improve] replace loop of progressbar in api/test. (#878)

2022-09-30 15:41:07 +09:00 · 2022-09-30 15:41:07 +09:00 · 1b4e9cd22a
parent 4eaaf89618
commit 1b4e9cd22a
1 changed files with 1 additions and 2 deletions
--- a/mmcls/apis/test.py
+++ b/mmcls/apis/test.py
@ -79,8 +79,7 @@ def single_gpu_test(model,
                    **show_kwargs)

        batch_size = data['img'].size(0)
-        for _ in range(batch_size):
-            prog_bar.update()
+        prog_bar.update(batch_size)
    return results