如何为我的张量流模型实现转移学习和微调？

Question

我有一个预先训练的模型（张量流模型），使用来自公开数据集的数据进行训练。我有元文件和ckpt文件。我想使用私有数据集中的新数据来训练我的张量流模型。我的数据集很小，因此我想根据“策略2”或“策略3”对模型进行微调。

• 策略2：训练一些图层，并冻结其他图层。
• 策略3：冻结卷积基础。

参考站点：https://towardsdatascience.com/transfer-learning-from-pre-trained-models-f2393f124751

但是，我找不到在张量流模型的转移学习和微调中实现的示例代码。 keras模型有很多例子。我如何在张量流模型的转移学习和微调中实现？

Answer 1

如果您不必使用Tensorflow的功能，也可以将示例代码与Tensorflow 2.0的tf.keras模块一起使用。