train Unet model with gray scale

[paulwu808]

Hello!
I wanna ask what code should i change if I wanna train a model with gray scale images

[glbessa]

Hello paulwu808, the module documentation has a section about it. You can find it here.