RGB dataset

[yug125lk]

Hi, thank you again. I used the code to train my custom dataset pair dataset; (input image, target image). It works when I convert the dataset to grayscale, but it gave me this error with RGB:

RuntimeError: Expected 3D (unbatched) or 4D (batched) input to conv2d, but got input of size: [10, 2, 128, 128, 3]