from local

This commit is contained in:
dohe0342 2023-01-03 13:34:50 +09:00
parent 6ffc0bcd5b
commit 78f955e57f
3 changed files with 3 additions and 1 deletions

View File

@ -77,5 +77,7 @@ class ConvolutionModule(nn.Module):
x = self.activation(self.norm(x))
x = self.pointwise_conv2(x)
x = x.transpose(1, 2)
x = self.layer_norm(x)
return x.transpose(1, 2)
return x