这周用ImageNet的例程,作者写这个原来的目的是:几十w张图片分1000类
boss硬要用我们的数据,改成用不到10w张图片,分2类。
先是在train_val.prototxt的fc8层把num_output改成2
但是loss值一直不收敛,一直是。。。#0: loss = 87.3365 (* 1 = 87.3365 loss)
accuracy也是随机水平,更扯淡的是,输出的predicted class一直是0
ToT
然后就在modelzoo里找原有的模型,硬加上fc8_new层。。。。。。
还是不收敛。
现在正在跑,GPU嗡嗡嗡叫的心好烦。。。
今晚肯定得加班测结果了。。。
唉,
有人能给点建议么
另外请问这口黑锅算谁的?
1
game115 2015-04-22 19:11:17 +08:00 via Android
solver里面把learning rate改小,每改1/10试试。还有gamma也要注意
|
2
scream7 2015-04-22 19:20:31 +08:00
那就先看深度学习,CNN啥的吧..
或者换成theano耍耍 |
3
livelxw 2015-04-22 19:29:00 +08:00 via iPhone
上theano辣
|