v5和v7的性能差异 - Githubissues

bubbliiiing / yolov7-pytorch

这是一个yolov7的库，可以用于训练自己的数据集。

GNU General Public License v3.0

861 stars 150 forks source link

Open HLiu1998 opened 1 year ago

HLiu1998 commented 1 year ago

大佬你好，我用预训练好的权重在Exdark数据集上微调，3000张训练图片，1800测试图片，epoch=120，冻结轮数为0，优化器是adam。但是在v7上的结果比在v5上的差，大概低了5个点。这是为什么呢？

bubbliiiing commented 1 year ago

都是Large模型？

HLiu1998 commented 1 year ago

都是Large模型？

是的，而且都是使用fp16，如果冻结50epoch的话，v7比v5低了两个点

bubbliiiing commented 1 year ago

可以试试SGD

HLiu1998 commented 1 year ago

好的，感谢