caffe finetune问题:按照网上的教程微调alexnet为什么loss一直是87.3365?
分类层的 num_output 和 标签的值域 不符合:a. 要知道imagenet是进行1000类的分类任务,我自己的数据是一个二分类,就一定要把最后‘fc8’InnerProduct的分类层的num_output: 2原来是1000,这个设成自己label的类别总数就可以。 b. 但是注意同时要修改train.prototxt和deploy.prototxt两个网络配置文件中的num_
原创
2017-08-08 17:25:08 ·
1005 阅读 ·
0 评论