While training dimension mismatch error is coming.

[kapilsingla22]

Hii @goutamyg
While training, On the first epoch only, I am getting the same type of error as mentioned here. Also after commenting the lines told by you, i am still getting the same error, can you help me with it. Thank you
RuntimeError: Given groups=1, weight of size [256, 128, 1, 1], expected input[128, 3, 128, 128] to have 128 channels, but got 3 channels instead