带有 InfoGainLoss 层 caffe 的错误 classification（完全是第一个 class）

Question

根据InfogainLoss layer 我已经从分支 https://github.com/shaibagon/caffe/tree/upgrade_infogain 替换了 SoftMaxWithLoss -> InfoGainLoss（它稳健地结合了 softmax 层和信息增益损失层）。

现在所有预测都是第一个 class。

有什么假设吗？

补充信息 solver, net, H 和 log my sover, net, generator H matrix, and log.

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

Answer 1

设置 base_lr: 0.001 而不是 0.01 解决了问题。