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.

Read the Docs v: stable
Versions
latest
stable
v0.10.2
v0.10.1
v0.10.0
v0.9.1
v0.9.0
v0.8.5
v0.8.4
v0.8.3
v0.8.2
v0.8.1
v0.8.0
v0.7.4
v0.7.3
v0.7.2
v0.7.1
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.