[Feature] Add autocast wrapper (#307)
* add autocast wrapper * fix docstring * fix docstring * fix compare version * fix unit test * fix incompatible arguments * fix as comment * fix unit test * rename auto_cast to autocast
Showing
- mmengine/optim/optimizer/amp_optimizer_wrapper.py 2 additions, 1 deletionmmengine/optim/optimizer/amp_optimizer_wrapper.py
- mmengine/runner/__init__.py 2 additions, 1 deletionmmengine/runner/__init__.py
- mmengine/runner/amp.py 87 additions, 0 deletionsmmengine/runner/amp.py
- mmengine/runner/loops.py 21 additions, 6 deletionsmmengine/runner/loops.py
- mmengine/runner/runner.py 8 additions, 4 deletionsmmengine/runner/runner.py
- tests/test_model/test_averaged_model.py 1 addition, 1 deletiontests/test_model/test_averaged_model.py
- tests/test_runner/test_amp.py 57 additions, 0 deletionstests/test_runner/test_amp.py
- tests/test_runner/test_runner.py 49 additions, 2 deletionstests/test_runner/test_runner.py
Loading
Please register or sign in to comment