A Problem When Run Tflite Model(the Result Of Tflite Model Is Nan)

I trained a model to convert sketch picture to color picture. enter image description here middle is ground truth, left is original and right is pridiction. This result was ran by

Solution 1:

I find a solution about this question. Just change the attribute fused of tf.keras.layers.bn to false. The output is correct.