训练好的在 runs/train/exp8/weights/last.pt 或者 best.pt 现在要再次训练600次 请问 在train.py中怎么设置?还有如果要继续训练的话 尺寸以及其他相关参数是不是要和之前的一样。
有两种做法。一种是在训练的时候加上 --resume这个参数,或者在代码里面将默认值改成True,然后将epoch在原来的基础之上加上600.如果这里报错:
直接将这里改成你上次训练停止的位置+1就行,至于上次停止的位置,你可以看下exp8文件下面的有个results.csv,里面的之后一行是多少直接+1就行。
另外一种做法就是将你上次训练好的模型作为预训练模型,epoch改成600,其他不动就行。