bubbliiiing / yolov5-pytorch

这是一个YoloV5-pytorch的源码,可以用于训练自己的模型。
GNU General Public License v3.0
967 stars 171 forks source link

swin_transformer #68

Open 9Godjasen opened 1 year ago

9Godjasen commented 1 year ago

从提供的swin_transformer权重开始训练感觉和没用预训练权重一样,前几个epoch精度基本都是0,up也是这样的吗?

chenzhaojun1996 commented 1 year ago

我前25个epoch,3个类别的recall均为0,有没有一样的

9Godjasen commented 1 year ago

我前25个epoch,3个类别的recall均为0,有没有一样的

一样

chenzhaojun1996 commented 1 year ago

你现在还在继续训练嘛

9Godjasen commented 1 year ago

你现在还在继续训练嘛

没有了,实验室设备不太行

chenzhaojun1996 commented 1 year ago

你现在还在继续训练嘛

没有了,实验室设备不太行

我v100训练80多epoch 还是不行 ,采用另外一个ConVext那个backbone也不行

9Godjasen commented 1 year ago

你现在还在继续训练嘛

没有了,实验室设备不太行 我v100训练80多epoch 还是不行 ,采用另外一个ConVext那个backbone也不行

我用swin训了300个epoch,ap50只有3左右,很离谱

chenzhaojun1996 commented 1 year ago

那你应该是没成功加载权重,训练初始化时候显示加载了多少层不

bubbliiiing commented 1 year ago

emm……swin效果很一般,如果用swin的话,建议调一下学习率或者优化器,用adam

9Godjasen commented 1 year ago

那你应该是没成功加载权重,训练初始化时候显示加载了多少层不

加载成功了,应该就是up说的学习率和优化器的问题。

bubbliiiing commented 1 year ago

嗯 adamw最好

nakuda1 commented 1 year ago

halo, 想问下现在你的swintransfomer有比较好的结果了嘛

pycxxxx commented 1 year ago

从提供的swin_transformer权重开始训练感觉和没用预训练权重一样,前几个epoch精度基本都是0,up也是这样的吗?

你好,我能请问一下swintransformer对应的是哪一个预训练权重文件吗

bubbliiiing commented 1 year ago

有个主干的权值哈,没有整个的