[Fix] gpu collection during evaluation (#1208)

open-mmlab · Jun 24, 2023 · 42dd97d · 42dd97d
1 parent 04b0ffe
commit 42dd97d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmengine/evaluator/metric.py b/mmengine/evaluator/metric.py
@@ -125,7 +125,7 @@ def evaluate(self, size: int) -> dict:
                 self.collect_device,
                 tmpdir=self.collect_dir)
         else:
-            collect_results(self.results, size, self.collect_device)
+            results = collect_results(self.results, size, self.collect_device)
 
         if is_main_process():
             # cast all tensors in results list to cpu