SSD训练ILSVRC2015,detection_eval一直很小,不到百分之一

数据使用的是ILSVRC2015 DET数据,有添加作者提到的ILSVRC2015_DET_fix.tar.gz,ILSVRC2015_DET_val1_val2.tar.gz。使用的是ssd_pascal.py,修改num_classes为201,直接跑,detection_eval会一直NAN,所以,先用base_lr=0.0001跑几百或几千次迭代,再基于此恢复训练,base_lr恢复为0.001。但几万或十几万次迭代后,detection_eval一直在0.0001~0.002左右飘动。
训练voc0712时是正常的,60k次迭代后达到0.70左右。
 
 
已邀请:

jiangqr

赞同来自: 辛淼 宇Never

原因已经找到,当时因为gpu显存不足,batch_size改为8,恢复原来的值就好了。

要回复问题请先登录注册