Colorful - Yellow results

[i-radwan]

Hi there,
I'm trying to train my own Colorful model to reproduce your results, however, no matter how long we train the model, the output images are always yellow colored.
Any suggestions?
Sample: here