From 9abe6109316bca240ca71a32806b1e6e39979ce0 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sun, 26 Jan 2020 11:33:31 -0800 Subject: [PATCH] Used wrong channel var for split --- timm/models/sknet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/sknet.py b/timm/models/sknet.py index 4bc2061d..d3a4fb5d 100644 --- 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]