RuntimeError: Given groups=1, weight of size [64, 64, 5, 5, 5], expected input[1, 16, 64, 8, 8] to have 64 channels, but got 16 channels instead
Hello i use pytorch to make a CNN with Cirfar data. And I got this error.
class Net(nn.Module): def init(self): super(Net, self).init() self.conv1 = nn.Conv2d( in_channels = 3, out_channels = 32, kernel_size = 5, stride = 1, padding = 2 ) self.conv2 = nn.Conv2d( in_channels=32, out_channels=64, kernel_size=5, stride=1, padding=2 ) self.conv3 = nn.Conv3d( in_channels=64, out_channels=64, kernel_size=5, stride=1, padding=2 ) self.pool = nn.MaxPool2d(2,2) self.fc1 = nn.Linear(1024, 0) self.fc2 = nn.Linear(0, 1024) self.fc3 = nn.Linear(1024, 10)
def forward(self, x):
x = self.pool(F.relu(self.conv1(x)))
x = self.pool(F.relu(self.conv2(x)))
print('x_shape:',x.shape)
x = self.pool(F.relu(self.conv3(x)))
x = torch.flatten(x, 1) # flatten all dimensions except batch
x = F.relu(self.fc1(x))
x = F.relu(self.fc2(x))
x = self.fc3(x)
return (x)
model = Net() count_parameters(model) model.to(device)
criterion = nn.CrossEntropyLoss() optimizer = optim.SGD(model.parameters(), lr=0.01)
source https://stackoverflow.com/questions/77671608/runtimeerror-given-groups-1-weight-of-size-64-64-5-5-5-expected-input1
Comments
Post a Comment