Update __init__.py

open-mmlab · Sep 4, 2023 · 60ec5b7 · 60ec5b7
1 parent 9f13bf8
commit 60ec5b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmengine/runner/__init__.py b/mmengine/runner/__init__.py
@@ -1,13 +1,13 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 from ._flexible_runner import FlexibleRunner
+from .activation_checkpointing import turn_on_activation_checkpointing
 from .amp import autocast
 from .base_loop import BaseLoop
 from .checkpoint import (CheckpointLoader, find_latest_checkpoint,
                          get_deprecated_model_names, get_external_models,
                          get_mmcls_models, get_state_dict,
                          get_torchvision_models, load_checkpoint,
                          load_state_dict, 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