mxnet运行时遇到问题及解决方法


1.训练好模型之后,进行预测时出现这种错误:

1 mxnet.base.MXNetError: [15:05:50] src/ndarray/ndarray.cc:239: Check failed: from.shape() == to->shape() operands shape mismatchfrom.shape = (63,) to.shape=(11,)

这种问题的解决方法,在全链接层fc中设置num_hidden你需要分类的个数和训练的模型的实际分类的个数参数不匹配,查看模型json,修改为对应值即可,这里修改成63及63个分类即可!


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM