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

open-mmlab · May 6, 2021 · db44d16 · db44d16
1 parent 4403923
commit db44d16
Showing 1 changed file with 1 addition and 1 deletion.
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()