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

来源:互联网 发布:淘宝返积分怎么领取 编辑:程序博客网 时间:2024/04/28 13:57

分类层的 num_output 和 标签的值域 不符合:

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

我按这个修改的,loss小了,之前也是87.3365

原创粉丝点击