mmyolo/docs/zh_cn/common_usage/amp_training.md

345 B
Raw Blame History

自动混合精度AMP训练

如果要开启自动混合精度AMP训练在训练命令最后加上 --amp 即可, 命令如下:

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

具体例子如下:

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