mirror of
https://github.com/open-mmlab/mmyolo.git
synced 2025-06-03 15:00:20 +08:00
43d97b76f6
* Add files via upload * cat_single_channel_test * [Feature] Add confusion matrix drawing tool * [Feature] Add confusion matrix drawing tool * [Feature] Add confusion matrix drawing tool * [Feature] Add confusion matrix drawing tool * [Docs] update amp_training.md, resume_training.md * [Docs] Update amp_training.md, resume_training.md * [Docs] Update amp_training.md, resume_training.md * [Docs] Update amp_training.md, resume_training.md * [Docs] Update amp_training.md, resume_training.md * [Docs] Update amp_training.md, resume_training.md
14 lines
375 B
Markdown
14 lines
375 B
Markdown
# Automatic mixed precision（AMP）training
|
||
|
||
To enable Automatic Mixing Precision (AMP) training, add `--amp` to the end of the training command, which is as follows:
|
||
|
||
```shell
|
||
python tools/train.py python ./tools/train.py ${CONFIG} --amp
|
||
```
|
||
|
||
Specific examples are as follows:
|
||
|
||
```shell
|
||
python tools/train.py configs/yolov5/yolov5_s-v61_syncbn_8xb16-300e_coco.py --amp
|
||
```
|