search

LSH9832 / edgeyolo

an edge-real-time anchor-free object detector with decent performance
Apache License 2.0
446 stars 59 forks source link

请问一下可以断点训练吗 #26

Open JerryLeeSir opened 1 year ago

JerryLeeSir commented 1 year ago

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

LSH9832 commented 1 year ago

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

weights: "./edgeyolo_coco.pth"

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

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