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. |