做一个10分类的问题,最后的全连接层的numout可以设成其他的吗?

发现最后的innerproduct层的numout不是10.。。。。这个是不是得重新训练啊?
已邀请:

B-Kalasiky

赞同来自:

不是10也能训练吗??(手动滑稽) 如果不想重新训练,微调不可以吗?

Evence

赞同来自:

做10分类的任务num_output应该是11,全连接层需要11个输出维度(num_output:11),还有一个是背景,不在你10个类别当中的,千万不要忘了哦,不然就出问题了,很大的问题。如果是11就对了,不是11就重新训练吧。

要回复问题请先登录注册