deepkashiwa20 / MepoGNN

[ECMLPKDD22] MepoGNN: Metapopulation Epidemic Forecasting with Graph Neural Networks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Error of data size when running the demo

TianH98 opened this issue · comments

Dear author of MepoGNN:

I'm trying to run the demo code. Hoewever I received an error "RuntimeError: Expected 2D (unbatched) or 3D (batched) input to conv1d, but got input of size: [32, 32, 47, 15]". Could you please help to take a look and modify your code?

Thank you very much:)

commented

@TianH98 Thank you for your interest. Sorry for the late reply.
I have run the code in my environment and it seems work without errors.
The error you met might be caused by the version differences of pytorch. The code is implemented with pytorch 1.9.0. If you used other versions of pytorch, the minor modifications of code might be needed.
(Possible solution: conv1d -> conv2d.)