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

Used wrong channel var for split

Browse Source
This commit is contained in:
Ross Wightman 2020-01-26 11:33:31 -08:00
parent 58e28dc7e7
commit 9abe610931
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/sknet.py
View File

@ -106,7 +106,7 @@ class SelectiveKernelConv(nn.Module):
def forward(self, x):
if self.split_input:
x_split = torch.split(x, self.out_channels // self.num_paths, 1)
x_split = torch.split(x, self.in_channels // self.num_paths, 1)
x_paths = [op(x_split[i]) for i, op in enumerate(self.paths)]
else:
x_paths = [op(x) for op in self.paths]