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.

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

error when run the GCN-M model #2

Open
kanghui-learning opened this issue Apr 28, 2024 · 0 comments
Open

error when run the GCN-M model #2

kanghui-learning opened this issue Apr 28, 2024 · 0 comments

Comments

@kanghui-learning
Copy link

In the definition of GCNMdynamic model, define the gated_conv as:
self.gate_convs.append(nn.Conv1d(in_channels=residual_channels, out_channels=dilation_channels, kernel_size=(1, kernel_size), dilation=new_dilation))

when run the model in metr_la dataset, get an error when calculate gate = self.gate_convs[i](residual) # kernel=(1,2)
RuntimeError: Expected 2D (unbatched) or 3D (batched) input to conv1d, but got input of size: [32, 32, 207, 13]

I think the residual got the correct shape of (B, residual_size, D, F) as noted. Is this a problem caused by the improper vision of torch?
Could you please help me solve this problem, I will appreciate any assistance you can provide.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant