mirror of
https://github.com/open-mmlab/mmengine.git
synced 2025-06-03 21:54:44 +08:00
26 lines
1.3 KiB
Python
26 lines
1.3 KiB
Python
# 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 .log_processor import LogProcessor
|
|
from .loops import EpochBasedTrainLoop, IterBasedTrainLoop, TestLoop, ValLoop
|
|
from .priority import Priority, get_priority
|
|
from .runner import Runner
|
|
from .utils import set_random_seed
|
|
|
|
__all__ = [
|
|
'BaseLoop', 'load_state_dict', 'get_torchvision_models',
|
|
'get_external_models', 'get_mmcls_models', 'get_deprecated_model_names',
|
|
'CheckpointLoader', 'load_checkpoint', 'weights_to_cpu', 'get_state_dict',
|
|
'save_checkpoint', 'EpochBasedTrainLoop', 'IterBasedTrainLoop', 'ValLoop',
|
|
'TestLoop', 'Runner', 'get_priority', 'Priority', 'find_latest_checkpoint',
|
|
'autocast', 'LogProcessor', 'set_random_seed', 'FlexibleRunner',
|
|
'turn_on_activation_checkpointing'
|
|
]
|