转载: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了。