Hello,
I am getting this error after first epoch. The model is getting trained for the very first epoch and then giving this error. I am doing segmentation with my images as .jpg files and masks as .png files. Please let me know the solution if any. Thanks in advance.
RuntimeError: Given groups=1, weight of size 64 3 3 3, expected input[1, 1, 480, 640] to have 3 channels, but got 1 channels instead