Sir! I think there are careless mistakes in 180 and 183 lines of train.py

The process of getting "predicted_RGB" and   "target_RGB" is wrong.
the operating will get "nan" matrix in second channel.
I think it is reason why our result deviates from the real image in hue, which is training model with real multispectral and panchromatic Images。