Training tends to yellow colors

[i-radwan]

Hi there,
Thanks for the code, I'm currently trying to train my own model using your Keras implementation, but the results always tend to be yellow colored (check this). I don't know if there's something wrong, can you please give me hints.