我有一个自己的训练模型(在 tf 1.14 中训练)，我用它来预训练另一个模型。因此，它不是您可以在 tf.keras.applications 中找到的知名模型之一，例如 ResNet 等。

我想将我的工作切换到 tf 2.0，但找不到任何链接来向我提供有关如何在 tf2 中加载 tf 1.14 检查点和图形的详细信息。

是否可以使用来自 tf 1 的模型信息在 tensorflow 2 中进行迁移学习？

最佳答案

如果你:

• 保存来自 tf 1.14 模型的权重 (old_model.save_weights(path))
• 在 2.0 中构建相同的模型(只是图层布局)(new_model = build_and_compile_model())
• 将 1.14 权重加载到 2.0 模型中 (new_model.load_weights(path/to/old_model_weights))

希望对你有帮助

https://stackoverflow.com/questions/58944755/