We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3044b1a commit 3ec5918Copy full SHA for 3ec5918
timm/models/layers/norm.py
@@ -50,7 +50,7 @@ def __init__(self, num_channels, eps=1e-6, affine=True):
50
51
def forward(self, x: torch.Tensor) -> torch.Tensor:
52
if self._fast_norm:
53
- x = fast_layer_norm(x, self.normalized_shape, self.weight, self.bias, self.eps)
+ x = fast_layer_norm(x, self.normalized_shape, self.weight, self.bias, self.eps)
54
else:
55
x = F.layer_norm(x, self.normalized_shape, self.weight, self.bias, self.eps)
56
return x
@@ -65,7 +65,7 @@ def __init__(self, num_channels, eps=1e-6, affine=True):
65
66
x = x.permute(0, 2, 3, 1)
67
68
69
70
71
x = x.permute(0, 3, 1, 2)
0 commit comments