我的目标是使用TensorFlow Estimator API对具有不同类数的任务微调downloaded model。
我目前正在按照建议的here进行以下操作:
WarmStartSettings
引用目录model_fn
中建立完全相同的模型(这需要导入最初用于创建模型的py文件)-在分类层中仅更改num_classes
。这很好用,但似乎有点复杂,因为所有必需的信息都应该在SavedModel
中可用。是否可以通过Estimator API / SavedModel
利用WarmStartSettings
进行迁移学习?