首页 > 代码库 > mxnet运行时遇到问题及解决方法

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

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