为文本分类预测加载 CNN 模型时出错

Question

我是卷积神经网络的新手，我基本上遵循这个例子：https://mxnet.incubator.apache.org/tutorials/nlp/cnn.html?highlight=convolutional 训练我的数据。

现在我运行并拥有 .params 和 .json 文件，我试图使用以下代码加载模型：

sym, arg_params, aux_params = mx.model.load_checkpoint('cnn', 9)
mod = mx.mod.Module(symbol=sym, context=mx.cpu(), label_names=None)
mod.bind(for_training=False, data_shapes=[('data', (1,51))], 
         label_shapes=mod._label_shapes)
mod.set_params(arg_params, aux_params, allow_missing=True)

但它总是显示如下错误：

MXNetError：运算符 reshape4 中的错误：[11:59:49] src/operator/tensor/./matrix_op-inl.h:182：检查失败：oshape.Size() == dshape.Size() 目标形状大小与源不同。目标：255000 来源：2550

我不确定我做错了哪一部分。

Answer 1

对于 data_shapes 它应该是：

data_shapes=[('data', (50,56))]

其中 50 是 batch_size，56 是 sentence_size / sequence_length。