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

Question

我训练了一个模型来将素描图片转换为彩色图片。 enter image description here 中间是基本事实，左边是原创，右边是预测。此结果由 model.h5 运行，但当我使用 model.tflite 运行程序时出现错误结果。

这是我的转换代码

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

这是 model.tflite 运行的结果

[[[[nan nan nan] ...

Answer 1

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