File "trainCNNmodel.py", line 358, in
outputs = net(inputs)
File "/N/u/user/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in call
result = self.forward(*input, *kwargs)
File "trainCNNmodel.py", line 171, in forward
x = self.pool(F.relu(self.conv1(x)))
File "/N/u/user/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in call
result = self.forward(input, **kwargs)
File "/N/u/user/anaconda3/lib/python3.7/site-packages/torch/nn/modules/conv.py", line 345, in forward
return self.conv2d_forward(input, self.weight)
File "/N/u/user/anaconda3/lib/python3.7/site-packages/torch/nn/modules/conv.py", line 342, in conv2d_forward
self.padding, self.dilation, self.groups)
RuntimeError: Given groups=1, weight of size 128 4 2 2, expected input[175, 3, 256, 256] to have 4 channels, but got 3 channels instead
srun: error: j-071: task 0: Exited with exit code 1
I would really appreciate any help that can be given!! Thank you!
File "trainCNNmodel.py", line 358, in
outputs = net(inputs)
File "/N/u/user/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in call
result = self.forward(*input, *kwargs)
File "trainCNNmodel.py", line 171, in forward
x = self.pool(F.relu(self.conv1(x)))
File "/N/u/user/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in call
result = self.forward(input, **kwargs)
File "/N/u/user/anaconda3/lib/python3.7/site-packages/torch/nn/modules/conv.py", line 345, in forward
return self.conv2d_forward(input, self.weight)
File "/N/u/user/anaconda3/lib/python3.7/site-packages/torch/nn/modules/conv.py", line 342, in conv2d_forward
self.padding, self.dilation, self.groups)
RuntimeError: Given groups=1, weight of size 128 4 2 2, expected input[175, 3, 256, 256] to have 4 channels, but got 3 channels instead
srun: error: j-071: task 0: Exited with exit code 1
I would really appreciate any help that can be given!! Thank you!