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

时间:2022-05-06
本文章向大家介绍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个分类即可!