yolov5 断点训练/继续训练

yolov5 断点训练/继续训练

断点训练:

python train.py --resume
automatically find latest checkpoint (searches yolov5/ directory)(未训练完最初设定的epochs)

继续训练:

python train.py --weights path/to/best.pt
weights后直接加需要继续训练的权重地址(已训练完最初设定的epochs,但认为还未完全收敛,在此基础上继续训练,需注意学习率的调整)

你可能感兴趣的:(深度学习,机器学习,pytorch)