mmyolo/docs/en/common_usage/amp_training.md

375 B
Raw Permalink Blame History

Automatic mixed precisionAMPtraining

To enable Automatic Mixing Precision (AMP) training, add --amp to the end of the training command, which is as follows:

python tools/train.py python ./tools/train.py ${CONFIG} --amp

Specific examples are as follows:

python tools/train.py configs/yolov5/yolov5_s-v61_syncbn_8xb16-300e_coco.py --amp