[Feature] Support torch.compile since PyTorch2.0 (#976)
* enable compile configurations to support torch.compile in Runner * enable compilation in train, val and test * fix as comments * add docstring to illustrate usage * minor refine error message * add unittests * fix ut skip * add logging message to inform users * compile `train_step`, `val_step`, `test_step` instead * fix as comments * revert to compile `train_step` only due to pt2 issue * add documentation about torch.compile
Showing
- docs/en/common_usage/speed_up_training.md 29 additions, 0 deletionsdocs/en/common_usage/speed_up_training.md
- docs/zh_cn/common_usage/speed_up_training.md 29 additions, 0 deletionsdocs/zh_cn/common_usage/speed_up_training.md
- mmengine/runner/runner.py 37 additions, 0 deletionsmmengine/runner/runner.py
- tests/test_runner/test_runner.py 55 additions, 1 deletiontests/test_runner/test_runner.py
Loading
Please register or sign in to comment