Pytorch model doesn't produce the intended output

[Samir55]

Hello authors,
I am encountering a problem that I couldn’t reproduce the same output by
training your model on the ILSVRC2012 training set. The model losses
(G_CE And G_entr) tend to stabilize around 3.1 across epochs. Also,
the colorized images tend to have a somehow yellowish color. I trained
the model for 6 epochs. I am using Python 3.5 and CUDA 9.1 with GTX
1060 GPU. Can you help me discover what’s going on? I am adding
here some output pictures and the loss graph. Thanks for your time and
efforts.

[richzhang]

I believe I did not do the initialization properly in the pytorch code. I am looking into it. Thanks for your patience.