I’m getting a very similar error.
“Given groups=1, weight of size [10, 3, 7, 7], expected input[1, 64, 64, 3] to have 3 channels, but got 64 channels instead”
I understand the input should be [batch, channels, height, width] but my environment returns me arrays of size (64, 64, 3). How do I make it (3, 64, 64), input it into the nn.Conv2d and get around this error?
Please help. @ptrblck