mmengine.runner¶
Runner¶
A training helper for PyTorch. |
|
A training helper for PyTorch. |
Loop¶
Base loop class. |
|
Loop for epoch-based training. |
|
Loop for iter-based training. |
|
Loop for validation. |
|
Loop for test. |
Checkpoints¶
A general checkpoint loader to manage all schemes. |
Find the latest checkpoint from the given path. |
|
Returns a dictionary containing a whole state of the module. |
|
Load checkpoint from a file or URI. |
|
Load state_dict to a module. |
|
Save checkpoint to file. |
|
Copy a model state_dict to cpu. |
AMP¶
A wrapper of |
Miscellaneous¶
A log processor used to format log information collected from |
|
Hook priority levels. |
Get priority value. |
|
Set random seed. |