Model FIT using Keras works well only on CPU, on GPU loss NaN

Question

I have prepared this code, for image colorization:

train = dataGenerator.flow_from_directory("simple_images/data/", target_size=(256, 256), batch_si