doesn't work with grayscale images

[aziabari]

the code produces error with grayscale images.
My input crop size is 400, and the number of input and output channel is set to 1, then I get this error:

RuntimeError: Given groups=1, weight of size [64, 1, 7, 7], expected input[0, 3, 406, 406] to have 1 channels, but got 3 channels instead

When I change the number of input/output channels to 3, then the error disappears.

It its worth noting, with the same training data using cyclegan code/repo, I use number of input/output channels equal to 1 and I don't get any error!

[ErikValle]

Trick: modify the value of a pixel to force the RGB settings. For example, replace the value of the first pixel with [0,255,0].

[7W7W7W]

Trick: modify the value of a pixel to force the RGB settings. For example, replace the value of the first pixel with [0,255,0].

您好，请问具体哪里修改呢？谢谢