『YOLOV5』| 一文搞定训练过程中的意外终止、以及想继续增加训练轮数!

情况一:意外训练中断(程序未训练完成,想完成目标训练轮数)

如果上次的指定训练轮次(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中),所以不用指定,就算指定了也没用。

情况二:自动训

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值