无论是caffe还是tensorflow都得注意最后输出维度的问题,是几类最后全连接层就输出几个神经元,尤其是在把其他网络直接拿过来用的时候,一定要注意到最后神经元个数与类别匹配:
tensorflow错误如下所示:
版权声明:本文为zxyhhjs2017原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。
无论是caffe还是tensorflow都得注意最后输出维度的问题,是几类最后全连接层就输出几个神经元,尤其是在把其他网络直接拿过来用的时候,一定要注意到最后神经元个数与类别匹配:
tensorflow错误如下所示: