ssd 在 coco 上训练,出现 loss 为 Nan,test map 过低的问题
Created by: Aneureka
训练的命令是
python3 -u train.py --dataset coco2017 --data_dir datasets/coco --batch_size 32 --pretrain
ed_model pretrained/ssd_mobilenet_v1_coco --learning_rate 0.001
加载的是官方的 mobileNet + ssd 与训练模型, 然后在 epoch == 4 的时候就出现 loss 为 Nan 的问题了
训练的时候 best_model 对应的 map 有 0.3 左右,但测试的时候只有 0.02 已经考虑到 91 -> 81 的问题...