Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

STLayer类中的filter_conv和gate_conv #4

Open
1229308805 opened this issue Nov 26, 2023 · 3 comments
Open

STLayer类中的filter_conv和gate_conv #4

1229308805 opened this issue Nov 26, 2023 · 3 comments

Comments

@1229308805
Copy link

STLayer类中的filter_conv和gate_conv 分别是2D卷积和1D卷积，在forward方法中他们都接受residual作为输入，这导致了报错 Expected 2D (unbatched) or 3D (batched) input to conv1d, but got input of size: [2, 40, 207, 13]， 请问作者遇到这个问题了吗，程序是否真的存在问题？
@LiuZH-19
Copy link
Owner

LiuZH-19 commented Dec 5, 2023

这跟pytorch的版本有关，我用的torch 版本为1.10.2。你也可以修改相应代码，来适配你的pytorch版本。
你可以参考这个issue区的讨论来解决这个问题：nnzhan/Graph-WaveNet#34

@1229308805
Copy link
Author

这跟pytorch的版本有关，我用的torch 版本为1.10.2。你也可以修改相应代码，来适配你的pytorch版本。 你可以参考这个issue区的讨论来解决这个问题：nnzhan/Graph-WaveNet#34

非常感谢您的解答~

@wdlovecjy
Copy link

换成了作者所提到的torch版本，并且python=3.6，确实解决了这个问题

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@1229308805 @LiuZH-19 @wdlovecjy