Skip to content

Commit

Permalink
[Fix] Fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
fanqiNO1 committed Feb 22, 2024
1 parent 850ae69 commit 14a86fb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mmengine/runner/loops.py
Original file line number Diff line number Diff line change
Expand Up @@ -400,7 +400,7 @@ def run_iter(self, idx, data_batch: Sequence[dict]):
'before_val_iter', batch_idx=idx, data_batch=data_batch)
# outputs should be sequence of BaseDataElement
with autocast(enabled=self.fp16):
outputs = self.runner.model.test_step(data_batch)
outputs = self.runner.model.val_step(data_batch)
if isinstance(outputs[-1],
BaseDataElement) and outputs.keys() == ['loss']:
loss = outputs[-1].loss # type: ignore
Expand Down

0 comments on commit 14a86fb

Please sign in to comment.