search

bubbliiiing / yolox-pytorch

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

训练coco数据集mAP偏低的问题 #120

Open YoungLNB opened 1 year ago

YoungLNB commented 1 year ago

您好，我采用的Adam的优化器，Init_Epoch = 0，UnFreeze_Epoch = 100，Freeze_Train = False，optimizer_type = 'adam'，Init_lr = 1e-3，weight_decay = 0。（不冻结）未加载预训练权重，训练的最好结果只有34.4。还望不吝赐教

bubbliiiing commented 1 year ago

鹅……100少了些吧……而且为什么要用adam

YoungLNB commented 1 year ago

我在训练VOC数据集时，未加载预训练权重，也是采用Init_Epoch = 0，UnFreeze_Epoch = 100，Freeze_Train = False，optimizer_type = 'adam'，Init_lr = 1e-3，weight_decay = 0。（不冻结）。 这个配置，训练出来的结果也较为合理，能够达到81.0%。同样的训练配置，COCO只能达到34.4%

bubbliiiing commented 1 year ago

COCO这个还是少了的，需要更大一些

YoungLNB commented 1 year ago

好的，明白了，多谢指点