首页 > Web开发 > 详细

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

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

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运行时遇到问题及解决方法

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!