edgeyolo icon indicating copy to clipboard operation
edgeyolo copied to clipboard

请问一下可以断点训练吗

Open JerryLeeSir opened this issue 2 years ago • 1 comments

我看train.py函数里只有两个参数选项,在训练开始时有一句no weight file found, setup models from cfg file /mnt/edgeyolo-main/params/model/edgeyolo.yaml是不是意味着可以加载之前的训练模型继续训练呀

JerryLeeSir avatar May 03 '23 23:05 JerryLeeSir

是的,在params/train/train_settings.yaml中第二个参数设置为

weights: "./edgeyolo_coco.pth"

便可使用这里提供的预训练权重开始训练,改为训练时的last.pth即可从断点继续训练

建议使用训练开始时使用我们提供的预训练权重以保证更好的效果

LSH9832 avatar May 04 '23 05:05 LSH9832