[Fix] Fix test

fanqiNO1 · Feb 22, 2024 · 0e7fcdc · 0e7fcdc
1 parent 5d1ecca
commit 0e7fcdc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mmengine/runner/loops.py b/mmengine/runner/loops.py
@@ -400,6 +400,7 @@ def run_iter(self, idx, data_batch: Sequence[dict]):
         # outputs should be sequence of BaseDataElement
         with autocast(enabled=self.fp16):
             results = self.runner.model.test_step(data_batch)
+        outputs, loss = list(), dict()
         if isinstance(results, tuple):
             outputs, loss = results
         elif isinstance(results, list):
@@ -494,6 +495,7 @@ def run_iter(self, idx, data_batch: Sequence[dict]) -> None:
         # predictions should be sequence of BaseDataElement
         with autocast(enabled=self.fp16):
             results = self.runner.model.test_step(data_batch)
+        outputs, loss = list(), dict()
         if isinstance(results, tuple):
             outputs, loss = results
         elif isinstance(results, list):