我如何在张量流模型的转移学习和微调中实现?

时间:2020-03-02 04:45:56

标签: python tensorflow transfer

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

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

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

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

1 个答案:

答案 0 :(得分:0)

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