RuntimeError: output with shape [1, 256, 256] doesn't match the broadcast shape [3, 256, 256]

[zqwang21]

Hi I get the following error when running the code:

I would appreciate your help on how to resolve this.

Thank you

[gibiee]

(Oh, it's too late...)

If you edit data_tools/data_loader_condition_final_new.py, it will be solved.
Add .convert('RGB') twice.