Given groups=1, weight of size 6 1 5 5, expected input[1, 6, 24, 24] to have 1 channels, but got 6 channels instead

1

class Network(nn.Module):
def init(self):
super().init()
self.conv1 = nn.Conv2d(in_channels=1, out_channels=6, kernel_size=5)
self.conv2 = nn.Conv2d(in_channels=6, out_channels=12, kernel_size=5)

    self.fc1 = nn.Linear(in_features=12*4*4, out_features=120)
    self.fc2 = nn.Linear(in_features=120, out_features=60)
    self.out = nn.Linear(in_features=60, out_features=10)
    
 

def forward(self,t):
    t = self.conv1(t)
    t = F.relu(self.conv1(t))
    t = F.max_pool2d(t, kernel_size=2, stride=2)
    
    t = self.conv2(t)
    t = F.relu(self.conv2(t))
    t = F.max_pool2d(t, kernel_size=2, stride=2)
    
    
    t = F.relu(self.fc1(t.reshape(-1, 12*4*4)))
    t = F.relu(self.fc2(t))
    t = self.out(t)
    
    return t

network = Network()
sample = next(iter(train_set))

image, label = sample
image.shape

image.unsqueeze(0).shape

pred = network(image.unsqueeze(0))

don’t know why I am getting this error

RuntimeError Traceback (most recent call last)
in
----> 1 pred = network(image.unsqueeze(0))

~\Anaconda3\lib\site-packages\torch\nn\modules\module.py in call(self, *input, **kwargs)
530 result = self._slow_forward(*input, **kwargs)
531 else:
–> 532 result = self.forward(*input, **kwargs)
533 for hook in self._forward_hooks.values():
534 hook_result = hook(self, input, result)

in forward(self, t)
13 def forward(self,t):
14 t = self.conv1(t)
—> 15 t = F.relu(self.conv1(t))
16 t = F.max_pool2d(t, kernel_size=2, stride=2)
17

~\Anaconda3\lib\site-packages\torch\nn\modules\module.py in call(self, *input, **kwargs)
530 result = self._slow_forward(*input, **kwargs)
531 else:
–> 532 result = self.forward(*input, **kwargs)
533 for hook in self._forward_hooks.values():
534 hook_result = hook(self, input, result)

~\Anaconda3\lib\site-packages\torch\nn\modules\conv.py in forward(self, input)
343
344 def forward(self, input):
–> 345 return self.conv2d_forward(input, self.weight)
346
347 class Conv3d(_ConvNd):

~\Anaconda3\lib\site-packages\torch\nn\modules\conv.py in conv2d_forward(self, input, weight)
340 _pair(0), self.dilation, self.groups)
341 return F.conv2d(input, weight, self.bias, self.stride,
–> 342 self.padding, self.dilation, self.groups)
343
344 def forward(self, input):

RuntimeError: Given groups=1, weight of size 6 1 5 5, expected input[1, 6, 24, 24] to have 1 channels, but got 6 channels instead

2

You are accidentally reusing self.conv1 (and self.conv2), which will throw this error, so you should remove the first usage.

1 Like