XudongLinthu / context-gated-convolution

Implementation for Context-Gated Convolution

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

runtime error

CR7forMadrid opened this issue · comments

Hello, I am getting the following error when I run your content gate conv. How do I fix this error?
return torch.matmul(out, x_un).view(b, self.oc, int(np.sqrt(l)), int(np.sqrt(l)))
RuntimeError: shape '[1, 64, 69, 69]' is invalid for input of size 306432