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

为什么保存模型后，再加载模型训练，giou变大了、map降低了 #23

Open

PL1296 opened this issue Aug 2, 2020 · 19 comments

PL1296 commented Aug 2, 2020

保存时giou=1.4，map=0.47，再训练一个epoch变2，map变0.364

Owner

WongKinYiu commented Aug 2, 2020

resume training is not yet supported.

xiaozhangya commented Nov 9, 2020

resume training is not yet supported.
现在支持了吗？加载模型训练重新训练？

Owner

WongKinYiu commented Nov 9, 2020

在這裡加上ema.updates = start_epoch * nb // accumulate就可以

xiaozhangya commented Nov 9, 2020 •

edited

Loading

在這裡加上ema.updates = start_epoch * nb // accumulate就可以

这个您试过吗？原理什么呢？另外有u3版本的加载位置吗？谢谢

xiaozhangya commented Nov 9, 2020

在這裡加上ema.updates = start_epoch * nb // accumulate就可以
下面两张图就是使用您说的方法重新加载模型，但是随着训练继续进行各项指标都在下降，很疑惑。模型能支持再次训练吗？这是一个很大的问题，做实验等于白做了

Owner

WongKinYiu commented Nov 9, 2020

u3版還沒時間整理新的程式碼

xiaozhangya commented Nov 9, 2020

u3版還沒時間整理新的程式碼

之前我问过您说的是u3、u5都是可用，我最近在做实验，发现了好多bug! 每次训练都会浪费很长时间呢。您能尽快解决u3这个bug吗？学了这么长时间了也不想再去另换其他版本的代码了。急求！麻烦您了

xiaozhangya commented Nov 9, 2020

保存时giou=1.4，map=0.47，再训练一个epoch变2，map变0.3

现在还在用这个版本的代码吗？

xiaozhangya commented Nov 9, 2020

u3版還沒時間整理新的程式碼

那u5版本加上那行确实可以重新加载保存的模型吗？如果可以就重新换这个版本学习了。

Owner

WongKinYiu commented Nov 9, 2020

可以呀
u3應該是加在192行您可以測試看看

xiaozhangya commented Nov 9, 2020 •

edited

Loading

可以呀
u3應該是加在192行您可以測試

u3版本就是加载到这一行呀
那这样都不行 u5可行吗现在没有多余的时间去整理这些原始模型上的bug了。

Owner

WongKinYiu commented Nov 9, 2020

u5不是preview版已測試可行 (u5_preview, u5_mish_preview未測試)
u3目前release的是preview版我這本機已經改版所以沒有測試過舊版

Owner

WongKinYiu commented Nov 9, 2020

因為不確定您的訓練流程
建議您把這裡先註解掉

xiaozhangya commented Nov 9, 2020

u5不是preview版已測試可行 (u5_preview, u5_mish_preview未測試)
u3目前release的是preview版我這本機已經改版所以沒有測試過舊版

您好我使用是这个版本的u3_preview的yolov4

xiaozhangya commented Nov 9, 2020

u5不是preview版已測試可行 (u5_preview, u5_mish_preview未測試)
u3目前release的是preview版我這本機已經改版所以沒有測試過舊版

您好我使用是这个版本的u3_preview的yolov4

到底u3版本和u5版本分别是那个可以使用？您如果更新了能分享到GitHub吗？另外刚才您说的注释，还是基于u5的，u3那一段代码不会被执行的。所以跟那一段代码似乎没有太大关系

xiaozhangya commented Nov 9, 2020

因為不確定您的訓練流程
建議您把這裡先註解掉

到底u3版本和u5版本分别是那个可以使用？您如果更新了能分享到GitHub吗？另外刚才您说的注释，还是基于u5的，u3那一段代码不会被执行的。所以跟那一段代码似乎没有太大关系

xiaozhangya commented Nov 9, 2020

因為不確定您的訓練流程
建議您把這裡先註解掉

还在吗？ u3版本和u5版本分别是那个可以使用？

xiaozhangya commented Nov 9, 2020

u3目前release的是preview版我這本機已經改版所以沒有測試過舊版

u3目前release的是preview版我這本機已經改版所以沒有測試過舊版，您是有最新的u3版本可以使用吗？能分享到github吗？谢谢啦！！

joe660 commented May 24, 2021

resume training is not yet supported.

U3_pre版本不支持保存的模型在训练但是您预训练的权重我却可以拿来直接使用。。

我假如在公开数据集上跑出一个预训练权重，然后再自己数据集上加载这个预训练权重运行可以吗？？？？
谢谢您的回复

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