这个错误提示看起来似乎是你在训练和测试时的 Batch Size 不一样。可以参考 https://tf.wiki/zh_hans/deployment/serving.html#keras 为@tf.function 设置 TensorSpec，使用 None 代表 Batch Size 大小可变。
这个错误提示看起来似乎是你在训练和测试时的 Batch Size 不一样。可以参考 https://tf.wiki/zh_hans/deployment/serving.html#keras 为@tf.function 设置 TensorSpec，使用 None 代表 Batch Size 大小可变。