昨天周日调好参数了,感觉大概能上到百分之90,我昨晚还特意看了看我们服务器的额度,看错了,以为用了33%结果最后是只剩下33%,训练一半,也就是今天早上八点,服务器出了预警,训练一半而中道报废,因此我想分享一下有关yolov5的断点训练的方法,这种方法可以做到接着之前的东西训练
步骤一
按下Ctrl+c 停止训练
步骤二
找到你当时的训练结果存储位置,一般是runs/train/exp/opt.yaml,我的是第36,因此我需要将runs/train/exp36/opt.yaml里面的resume后面改为true,

之后再将再将train.py里面的resume中default改为true

步骤三
直接执行原来命令即可
python train.py --weights yolov5x6.pt --img 1536 --data data_x6_0.9.yaml --batch-size 6 --device 0 --epochs 200
浙公网安备 33010602011771号