码迷,mamicode.com
首页 > Web开发 > 详细

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

时间:2017-01-09 16:32:36      阅读:592      评论:0      收藏:0      [点我收藏+]

标签:check   pre   net   sha   ase   .sh   fail   匹配   预测   

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个分类即可!

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

标签:check   pre   net   sha   ase   .sh   fail   匹配   预测   

原文地址:http://www.cnblogs.com/gongxijun/p/6265373.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!