Caffe训练分类器不收敛,loss居高不下

来源:互联网 发布:js实现日期格式化 编辑:程序博客网 时间:2024/06/06 05:09
转载:http://blog.sina.com.cn/s/blog_141f234870102w941.html
我利用Caffe训练一个基于AlexNet三分类分类器,将train_val.prototxt全连接输出层输出类别数目改为3,训练一直不收敛,loss很高;当把输出改成4或1000(>3)的时候,网络可以收敛。也就是caffenet结构输出层类别数一定要大于我训练类别数才可以收敛!后来查了半天才发现原因,让我泪奔。。。
原来我把图像类别label设置成1,2,3,改成0,1,2后,最后全连接层转出改成3就OK了。
原创粉丝点击