search

WongKinYiu / PyTorch_YOLOv4

PyTorch implementation of YOLOv4
1.87k stars 585 forks source link

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

Open PL1296 opened 4 years ago

PL1296 commented 4 years ago

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

WongKinYiu commented 4 years ago

resume training is not yet supported.

xiaozhangya commented 3 years ago

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

WongKinYiu commented 3 years ago

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

xiaozhangya commented 3 years ago

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

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

xiaozhangya commented 3 years ago

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

WongKinYiu commented 3 years ago

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

xiaozhangya commented 3 years ago

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

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

xiaozhangya commented 3 years ago

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

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

xiaozhangya commented 3 years ago

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

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

WongKinYiu commented 3 years ago

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

xiaozhangya commented 3 years ago

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

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

WongKinYiu commented 3 years ago

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

WongKinYiu commented 3 years ago

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

xiaozhangya commented 3 years ago

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

您好 我使用是这个版本的u3_preview的yolov4
1604929058(1)

xiaozhangya commented 3 years ago

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

您好 我使用是这个版本的u3_preview的yolov4 1604929058(1)

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

xiaozhangya commented 3 years ago

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

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

xiaozhangya commented 3 years ago

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

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

xiaozhangya commented 3 years ago

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

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

joe660 commented 3 years ago

resume training is not yet supported.

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

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