hi this is my first post, so if not suitable sorry.
basically I am developing a NN that uses images. However, I receive this error:
RuntimeError: Given groups=1, weight[5, 3, 436, 436], so expected input[5, 436, 436, 3] to have 3 channels, but got 436 channels instead
I got that legent of weight[batch,channel,h,w], but input has this data shift.
I copied the sample code from this.
only change I made is just changing self.conv1 = nn.Conv2d(3,6,5) to nn.Conv2d(3, 5, 436)
I can provide more info if necessary. Besides, can I use h!=w input images? thanks!