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.