bubbliiiing / yolox-pytorch

这是一个yolox-pytorch的源码,可以用于训练自己的模型。
Apache License 2.0
877 stars 184 forks source link

网络不收敛 #125

Open 2292301049 opened 1 year ago

2292301049 commented 1 year ago

博主,我用你的预训练权重训练了YOLOX-l模型20多个epoch之后,loss不降反增是个什么情况 image

bubbliiiing commented 1 year ago

鹅,因为这个模型,只需要拟合cls,学习率上升后反而loss变大

2292301049 commented 1 year ago

鹅,因为这个模型,只需要拟合cls,学习率上升后反而loss变大

学习率用的是你复现的代码里的,没做什么改动,就是调整了一下batchsize

bubbliiiing commented 1 year ago

没说是你的问题,鹅……第一个epoch已经拟合了

2292301049 commented 1 year ago

没说是你的问题,鹅……第一个epoch已经拟合了

请问博主有什么方法可以解决吗,学习率也调整过了,没有什么效果

bubbliiiing commented 1 year ago

鹅……不载入头部就行

2292301049 commented 1 year ago

鹅……不载入头部就行

我是个菜鸟,咋能不载入头部,求指教

2292301049 commented 1 year ago

鹅……不载入头部就行

先采用冻结训练不知道是否可行

bubbliiiing commented 1 year ago

鹅,你把 self.head = YOLOXHead(num_classes, width, depthwise=depthwise) self.head改个名字,比如self.head1就行应该