• GitHub
  • Docs
    MMEngine

    MMCV

    MMEval

    MIM

    MMAction2

    MMPreTrain

    MMDetection

    MMDetection3D

    MMagic

    MMOCR

    MMPose

    MMSegmentation

    MMTracking

    MMFlow

    MMFewShot

    MMHuman3D

    MMRazor

    MMDeploy

    MMRotate

    MMYOLO

    OpenMMLab
    Homepage

    Open Platform

    GitHub

    Twitter

    Zhihu

    Shortcuts

    mmengine.runner

    Runner

    Runner

    A training helper for PyTorch.

    FlexibleRunner

    A training helper for PyTorch.

    Loop

    BaseLoop

    Base loop class.

    EpochBasedTrainLoop

    Loop for epoch-based training.

    IterBasedTrainLoop

    Loop for iter-based training.

    ValLoop

    Loop for validation.

    TestLoop

    Loop for test.

    Checkpoints

    CheckpointLoader

    A general checkpoint loader to manage all schemes.

    find_latest_checkpoint

    Find the latest checkpoint from the given path.

    get_deprecated_model_names

    get_external_models

    get_mmcls_models

    get_state_dict

    Returns a dictionary containing a whole state of the module.

    get_torchvision_models

    load_checkpoint

    Load checkpoint from a file or URI.

    load_state_dict

    Load state_dict to a module.

    save_checkpoint

    Save checkpoint to file.

    weights_to_cpu

    Copy a model state_dict to cpu.

    AMP

    autocast

    A wrapper of torch.autocast and toch.cuda.amp.autocast.

    Miscellaneous

    LogProcessor

    A log processor used to format log information collected from runner.message_hub.log_scalars.

    Priority

    Hook priority levels.

    get_priority

    Get priority value.

    set_random_seed

    Set random seed.

  • GitHub