mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

One too many changes at a time, fix missing C

Browse Source
This commit is contained in:
Ross Wightman 2020-04-01 13:19:08 -07:00 committed by Chris Ha
parent f17b42bc33
commit 1a9ab07307
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/models/layers/blurpool.py
View File

@ -49,7 +49,7 @@ class BlurPool2d(nn.Module):
self.blur_filter = fn(self.blur_filter)
def forward(self, input_tensor: torch.Tensor) -> torch.Tensor: # type: ignore
C = input_tensor.shape[1]
return F.conv2d(
self.padding(input_tensor),
self.blur_filter.type(input_tensor.dtype).expand(C, -1, -1, -1),
stride=self.stride, groups=input_tensor.shape[1])
self.blur_filter.type(input_tensor.dtype).expand(C, -1, -1, -1), stride=self.stride, groups=C)