TensorFlow:使用SavedModel通过WarmStartSettings通过不同数量的类进行迁移学习

时间:2018-12-06 05:22:15

标签: python tensorflow tensorflow-estimator transfer-learning

我的目标是使用TensorFlow Estimator API对具有不同类数的任务微调downloaded model

我目前正在按照建议的here进行以下操作:

  • 下载模型检查点并通过WarmStartSettings引用目录
  • model_fn中建立完全相同的模型(这需要导入最初用于创建模型的py文件)-在分类层中仅更改num_classes

这很好用,但似乎有点复杂,因为所有必需的信息都应该在SavedModel中可用。是否可以通过Estimator API / SavedModel利用WarmStartSettings进行迁移学习?

0 个答案:

没有答案