New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open

JerryLeeSir opened this issue May 3, 2023 · 1 comment

JerryLeeSir commented May 3, 2023

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

Owner

LSH9832 commented May 4, 2023 •

edited

Loading

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

weights: "./edgeyolo_coco.pth"

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

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment