mmengine.runner — mmengine 0.10.7 documentation (original) (raw)

mmengine.runner

Runner

Loop

Checkpoints

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.