tensorflow - 在 tf2.3 上训练的模型在 tf2.6 上表现不佳

Question

我已经在 tensorflow 2.3 上训练了一个模型，现在正在尝试对 tf2.6 进行推理。该模型最初是在 AWS 上训练的，因此 CUDA 和 CUDNN 版本以及环境的其余部分（python 3.7.11 和 3.8.12、numpy 1.18.5 和 numpy 1.19.5）是不同的。

模型完全相同，权重也相同。然而，对相同输入数据的推断会产生两个不同的向量。它们不一样，差别很大。

是什么导致了这个问题，我如何使结果相同？

标签： tensorflowkeras

Answer 1

不是解决方案，但找到了解决方法。似乎是 keras 2.6.0 有问题。从 tf2.6 降级到 tf2.5 就可以了，因为结果是一样的。