您的位置:首页 > Web前端

caffe finetune问题:按照网上的教程微调alexnet为什么loss一直是87.3365?

2017-08-08 17:25 405 查看
分类层的 num_output 和 标签的值域 不符合:

a. 要知道imagenet是进行1000类的分类任务,我自己的数据是一个二分类,就一定要把最后‘fc8’InnerProduct的分类层的num_output: 2原来是1000,这个设成自己label的类别总数就可以。 b. 但是注意同时要修改train.prototxt和deploy.prototxt两个网络配置文件中的num_output

我按这个修改的,loss小了,之前也是87.3365
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐