[Fix] Fix scale_lr in SingleDeviceStrategy

open-mmlab · Nov 20, 2023 · c558951 · c558951
1 parent 30262f4
commit c558951
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmengine/runner/_flexible_runner.py b/mmengine/runner/_flexible_runner.py
@@ -1177,7 +1177,7 @@ def train(self) -> nn.Module:
             max_epochs=self.max_epochs,
             max_iters=self.max_iters,
             train_micro_batch_size_per_gpu=_get_batch_size(
-                self._train_dataloader))  # type: ignore
+                self.train_dataloader))  # type: ignore
 
         self.strategy.prepare(
             self.model,