Wrong convolution

[846468230]

The convolution method in function forward is wrong, x=x.permute(0,2,1)

[yiyunzhao]

accordingly, the in-channel should be embedding size as well.