from local
This commit is contained in:
parent
6ffc0bcd5b
commit
78f955e57f
Binary file not shown.
Binary file not shown.
@ -77,5 +77,7 @@ class ConvolutionModule(nn.Module):
|
|||||||
x = self.activation(self.norm(x))
|
x = self.activation(self.norm(x))
|
||||||
|
|
||||||
x = self.pointwise_conv2(x)
|
x = self.pointwise_conv2(x)
|
||||||
|
x = x.transpose(1, 2)
|
||||||
|
x = self.layer_norm(x)
|
||||||
|
|
||||||
return x.transpose(1, 2)
|
return x
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user