Update runner.py

open-mmlab · Sep 4, 2023 · 1c59213 · 1c59213
1 parent 60ec5b7
commit 1c59213
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmengine/runner/runner.py b/mmengine/runner/runner.py
@@ -41,11 +41,11 @@
 from mmengine.utils.dl_utils import (TORCH_VERSION, collect_env,
                                      set_multi_processing)
 from mmengine.visualization import Visualizer
+from .activation_checkpointing import turn_on_activation_checkpointing
 from .base_loop import BaseLoop
 from .checkpoint import (_load_checkpoint, _load_checkpoint_to_model,
                          find_latest_checkpoint, save_checkpoint,
                          weights_to_cpu)
-from .activation_checkpointing import turn_on_activation_checkpointing
 from .log_processor import LogProcessor
 from .loops import EpochBasedTrainLoop, IterBasedTrainLoop, TestLoop, ValLoop
 from .priority import Priority, get_priority