使用InfoGainLoss图层caffe进行错误分类(完全是第1类)

时间:2016-05-27 20:07:50

标签: caffe

根据InfogainLoss layer 我已经取代了SoftMaxWithLoss - >来自分支https://github.com/shaibagon/caffe/tree/upgrade_infogain的InfoGainLoss(强大地结合了softmax层和infogain损失层)。

现在所有的预测都只是第一类。

任何假设?

其他信息 求解器,网络,H和记录我的sover,net,generator H matrix和log。

https://drive.google.com/a/smedx.com/file/d/0B4lunYl8YWUOQ3U3NzN6Tll5NEE/view?usp=sharing

1 个答案:

答案 0 :(得分:0)

设置base_lr: 0.001而非0.01可以解决问题。