ysn2233 / attentioned-dual-stage-stock-prediction Goto Github PK

老哥，能不能在代码中对照着论文大概解释一下两个阶段的Attention啊？

这种写法，只适合外部驱动序列只有一个的情况，如果多个，模型要修改的地方很多，不适合扩展性。

When I ran the code in test phase, I got this error

Traceback (most recent call last):
File "trainer.py", line 152, in
trainer.test(mname, batch_size)
File "trainer.py", line 66, in test
y_pred_train = self.predict(x_train, y_train, y_seq_train, batch_size)
File "trainer.py", line 90, in predict
y_res = self.decoder(code, var_y_input)
File "/home/tatsuhiro/PycharmProjects/untitled/venv/lib/python3.5/site-packages/torch/nn/modules/module.py", line 491, in call
result = self.forward(*input, **kwargs)
File "/home/tatsuhiro/PycharmProjects/untitled/attentioned-dual-stage-stock-prediction-master/model.py", line 101, in forward
ct = torch.bmm(beta_t.unsqueeze(1), h)
RuntimeError: invalid argument 6: wrong matrix size at /pytorch/aten/src/THC/generic/THCTensorMathBlas.cu:477

I do not change any of programs.
How can I solve this problem?