有人玩过 caffe 框架么?唉最近公司玩这个,我全是眼泪

2015-04-22 17:59:23 +08:00
 northisland

这周用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嗡嗡嗡叫的心好烦。。。
今晚肯定得加班测结果了。。。
唉,
有人能给点建议么
另外请问这口黑锅算谁的?

2124 次点击
所在节点    机器学习
3 条回复
game115
2015-04-22 19:11:17 +08:00
solver里面把learning rate改小,每改1/10试试。还有gamma也要注意
scream7
2015-04-22 19:20:31 +08:00
那就先看深度学习,CNN啥的吧..
或者换成theano耍耍
livelxw
2015-04-22 19:29:00 +08:00
上theano辣

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/185534

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX