Hi,In your paper the generator loss function is huber loss but in your pytorch version, the loss function is L1 loss

[meiguoofa]

So the L1 loss function more suit for Colorization task?