Xiaobo

> > 跑了下这个代码，请问预测时如何设置batch大小，我用训练的数据测试预测，当我要预测的数据不是按照batch大小输入时，会报下面的错误： > > RuntimeError(msg.format(expected_hidden_size, tuple(hx.size()))) > > RuntimeError: Expected hidden[0] size (2, 12, 100), got (2, 128, 100) > > 我用的训练的数据，只不过把batch修改成了12。 > > 您好，请问您这个问题解决了吗 你好，我觉得是在attention内部写死了batch size，目前还没能直接修改attention的代码，我用了个笨方法，就是加载模型后修改batch size为1，att_weight和relation_bias也全都是1个batch size，这样可以预测一条数据。代码如下：...