情况一:意外训练中断(程序未训练完成,想完成目标训练轮数)
如果上次的指定训练轮次(epochs)没跑完就意外中断了,直接在在命令行中加上 --resume + 上一次训练中last.pt的路径
例如:
python train.py --resume runs/exp2/weights/last.pt
也可以只写–resume后面不跟路径参数,程序会自动找到runs目录下所有exp/weights目录下的文件修改时间最晚的last.pt
python train.py --resume
而其他所有参数(–epochs、–batch-size等等)都会自动读取跟上次一样的(保存在runs/exp2/opt.yaml中),所以不用指定,就算指定了也没用。