bubbliiiing / mobilenet-yolov4-pytorch

这是一个mobilenet-yolov4的库,把yolov4主干网络修改成了mobilenet,修改了Panet的卷积组成,使参数量大幅度缩小。
MIT License
368 stars 103 forks source link

mobilenetv3未使用预训练权重出现问题 #13

Open HotPy opened 3 years ago

HotPy commented 3 years ago

大神您好,我没有使用您提供mobilenetv3预训练权重,而是采用mobilenetv3网络中自带的初始化方法进行训练,采用adadelta优化器,batchsize=16,训练集为voc的train07+12,在训练至360个epoch后,出现如下图的情形,请问有什么办法让网络继续收敛吗? epoch_loss_2021_06_14_05_16_44

bubbliiiing commented 3 years ago

这个mAP有多少啊…或者换个优化器会好一些

HotPy commented 3 years ago

当loss为8.527时,在test07上的mAP是67.41%,目前正在用Adam训练,想请教一下大神在训练的时候参数是怎么设定,以及loss最终会下潜到多少。

bubbliiiing commented 3 years ago

val_loss好像是到8左右

bubbliiiing commented 3 years ago

我一般用默认的参数