query2labels icon indicating copy to clipboard operation
query2labels copied to clipboard

关于训练过程,请问是正常的吗?

Open ShawnALiu opened this issue 1 year ago • 0 comments

backbone使用的是 swin_B_224_22k,在coco2014上做的训练,下面是其他训练参数: --batch-size 256
--gamma_pos 0 --gamma_neg 2 --dtgfl
--epochs 80
--lr 1e-5
--optim AdamW
--pretrained
--weight-decay 1e-2
--cutout --n_holes 1 --cut_fact 0.5
--hidden_dim 1024 --dim_feedforward 8192
--enc_layers 1 --dec_layers 2 --nheads 4 --amp

训练结果: best mAP 85.41201044291725 in ep 19 best regular mAP 85.41201044291725 in ep 19

我的训练过程是这样的,请问大佬,正常吗?最好的mAP是在常规模型上产生的,不是ema参数模型产生的,但是下图中,常规模型val_loss问题有点大,ema的loss和mAP倒是能接受。 大佬能帮忙解答一下吗?

训练过程

ShawnALiu avatar Mar 08 '24 08:03 ShawnALiu