1.使用111张图片,训练1000次,batch-size=4,lr=0.001
2.进行数据增广,对图片进行旋转90/180/270/360,共444张图片。gup上训练出现有时候会出现nan情况,猜想是数据集的原因。跳转lr=0.0001有时候仍然会出现nan。cpu上训练未出现nan,但Map反而下降。
3.减少数据增广的图片,仅使用原图和旋转90度的图片共222张。参数不变,map介于两者中间。
原文:https://www.cnblogs.com/lian-dong/p/12657924.html