如何将学习从 tensorflow 1.14 转移到 tf 2？

Question

我有自己的训练模型（在 tf 1.14 中训练），我用它来预训练另一个模型。因此，它不是您可以在tf.keras.applications 中找到的 ResNet/etc 等众所周知的模型之一。

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

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

Answer 1

如果你应该可以：

• 保存来自 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))

希望对你有帮助