请问有尝试过将该模型保存为savedmodel格式的吗？我保存成这样后再读取就会出错

[kill2013110]

使用savedmodel是因为要后续准备冻结模型。 函数用的tf.keras.model.save_model()和tf.keras.model.load_model() 分析了一下，是因为模型中包含了自定义的Lambda层，所以报错了， 对于lambda层我准备转为tf.function, 应该可以解决问题， 您有什么好的建议吗

[bubbliiiing]

我没有试过其实…

[Duton]

请问你解决这个问题了么？

[kill2013110]

请问你解决这个问题了么？

解决了，因为把loss当做自定义的lambda层也放进网络里了，这样训练时网络的输出就是loss，比较简洁，你可以仔细看训练模型的代码，但这个lambda层是不支持savedmodel格式的。 解决办法：可以建一个不包括自定义loss层的模型，然后将训练好的weight load进去，再savedmodel保存

[bubbliiiing]

，可以在预测的时候……save