infer batch size using len(result) in test function (#532)

Wuziyi616 · web-flow · commit db44d16e023f · 2021-05-06T08:16:46.000-07:00
diff --git a/mmseg/apis/test.py b/mmseg/apis/test.py
@@ -149,7 +149,7 @@ def multi_gpu_test(model,
             results.append(result)
 
         if rank == 0:
-            batch_size = data['img'][0].size(0)
+            batch_size = len(result)
             for _ in range(batch_size * world_size):
                 prog_bar.update()
 
