from local

This commit is contained in:
dohe0342 2023-01-03 13:37:07 +09:00
parent 27a42630c6
commit 98748964b4
2 changed files with 2 additions and 0 deletions

View File

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