invalid argument

[jiayi797]

Hello everyone,

when I run the code in my cpu computer, I run :

python main.py --horizon 24 --data data/electricity.txt --save save/elec.pt --output_fun Linear

Then it told me:

Traceback (most recent call last):
  File "main.py", line 149, in <module>
    train_loss = train(Data, Data.train[0], Data.train[1], model, criterion, optim, args.batch_size)
  File "main.py", line 55, in train
    output = model(X);
  File "/home/jiayi/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/jiayi/LSTNet/models/LSTNet.py", line 54, in forward
    s = s.view(batch_size, self.hidC, self.pt, self.skip);
RuntimeError: invalid argument 2: size '[128 x 100 x 6 x 24]' is invalid for input with 2073600 elements at /pytorch/aten/src/TH/THStorage.c:41

It puzzled me for whole day... Anyone meet this error? could you share solution with me?

Thank you anyway.

[holyglenn]

Did you get a change to fix this issue? @jiayi797

[jiayi797]

Unfortunately not. :( @holyglenn

[holyglenn]

I resolved it. Probably all you need is cast one of the float point number size to int. That's what I did.

…

On Tue, Jul 31, 2018 at 6:48 AM 贾已 ***@***.***> wrote: Unfortunately not. :( — You are receiving this because you commented. Reply to this email directly, view it on GitHub <#2 (comment)>, or mute the thread <https://github.com/notifications/unsubscribe-auth/AB0XVC4AMjbayaFsB7-VgIaLoDTsty3Nks5uMDXzgaJpZM4UTG-X> .

-- Yu Weiren

[laiguokun]

@holyglenn is right. Simply converts it to Int will resolve this.