训练的时候,发现epochs设置多了,训练中途发现收敛效果还可以,不用继续跑太多轮,于是想缩减epochs。但没找到解决的帖子…
修改步骤:
1.首先train文件的参数改一下:
- opochs改成减少后的轮数,比如先前是200,这次改成100
- model路径改成上次训练的last.pt
- 为了防止路径错误,建议全部使用绝对路径,然后把"\“都换成”"
from ultralytics import YOLO
if __name__ == '\_\_main\_\_':
model = YOLO("runs\\detect\\train\\weights\\last.pt")
results = model.train(data="C:\\Users\\Administrator\\Desktop\\ultralytics-main\\ultralytics-main\\ultralytics\\datasets\\mask\\data.yaml", epochs=100, batch=4, workers=2, resume=True, device=0)
2. 在trainer.py的__init__() 构造函数里,先用一个变量接收config的epochs
def \_\_init\_\_(self, cfg=DEFAULT_CFG, overrides=None, _callbacks=None):