Error when change num_hidden = 256

[data-compute-geek]

In using char_lstm.ipyn I changed num_hidden from 512 to 256, and the model training finished without errors. But I got the following error message when I tried to make the inference model using the following command.

model_inf = rnn_model.LSTMInferenceModel(
num_lstm_layer,
len(vocab) + 1,
num_hidden=num_hidden,
num_embed=num_embed,
num_label=len(vocab) + 1,
arg_params=arg_params,
ctx=mx.gpu(),
dropout=0.2)

[22:45:14] /home/ec2-user/src/mxnet/dmlc-core/include/dmlc/logging.h:235: [22:45:14] src/ndarray/ndarray.cc:231: Check failed: from.shape() == to->shape() operands shape mismatch
Traceback (most recent call last):
File "", line 10, in
File "/home/ec2-user/mxnet-notebooks/python/tutorials1/rnn_model.py", line 34, in init
arg_params[key].copyto(self.executor.arg_dict[key])
File "/usr/local/lib/python2.7/site-packages/mxnet-0.7.0-py2.7.egg/mxnet/ndarray.py", line 533, in copyto
return _internal._copyto(self, out=other)
File "/usr/local/lib/python2.7/site-packages/mxnet-0.7.0-py2.7.egg/mxnet/ndarray.py", line 1225, in unary_ndarray_function
c_array(ctypes.c_char_p, [c_str(str(i)) for i in kwargs.values()])))
File "/usr/local/lib/python2.7/site-packages/mxnet-0.7.0-py2.7.egg/mxnet/base.py", line 77, in check_call
raise MXNetError(py_str(_LIB.MXGetLastError()))
mxnet.base.MXNetError: [22:45:14] src/ndarray/ndarray.cc:231: Check failed: from.shape() == to->shape() operands shape mismatch

[data-compute-geek]

Re-posted the issue to the main thread.