Used wrong channel var for split

4 years ago · 9abe610931
parent 58e28dc7e7
commit 9abe610931
1 changed files with 1 additions and 1 deletions
--- a/timm/models/sknet.py
+++ b/timm/models/sknet.py
@ -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]