Python 运行tflite模型时出现问题（tflite模型的结果为nan）

我训练了一个模型将素描图片转换成彩色图片。中间是基本真理，左边是原创，右边是预言。此结果由model.h5运行，但当我使用model.tflite运行程序时，出现了错误的结果

这是我的转换代码

tflite_convert --keras_model_file=G:/pix2pix/generator.h5 --output_file=G:/pix2pix/convert.tflite

这是model.tflite运行的结果

[[[[nan nan nan] ...

---

我找到了解决这个问题的办法。只需将tf.keras.layers.bn的属性fused更改为false。输出是正确的。

我找到了这个问题的解决方案。只需将tf.keras.layers.bn的属性fused更改为false。输出是正确的