Skip to content

Commit 1a9ab07

Browse files
rwightmanchris-ha458
authored andcommitted
One too many changes at a time, fix missing C
1 parent f17b42b commit 1a9ab07

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

timm/models/layers/blurpool.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def _apply(self, fn):
4949
self.blur_filter = fn(self.blur_filter)
5050

5151
def forward(self, input_tensor: torch.Tensor) -> torch.Tensor: # type: ignore
52+
C = input_tensor.shape[1]
5253
return F.conv2d(
5354
self.padding(input_tensor),
54-
self.blur_filter.type(input_tensor.dtype).expand(C, -1, -1, -1),
55-
stride=self.stride, groups=input_tensor.shape[1])
55+
self.blur_filter.type(input_tensor.dtype).expand(C, -1, -1, -1), stride=self.stride, groups=C)

0 commit comments

Comments
 (0)