Given groups=1, weight[16, 1, 5, 5], so expected input[100, 3, 64, 64] to have 1 channels, but got 3 channels instead

Yeah. I just started learning pytorch.

I encourage you to do the tutorials provided by pytorch, they are quite good. https://pytorch.org/tutorials/

1 Like

Sure. Thanks a lot. Will check it out