从Tensorflow SavedModel格式转移学习

Question

假设我正在这样读取SavedModel格式的模型：

graph_definition = tf.saved_model.loader.load(
                                               current_session,
                                               [tf.saved_model.tag_constants.SERVING],
                                               path_to_model
                                           )

有没有一种方法可以自定义SavedModel以将其用于转移学习？我希望能够用新的层替换最终的分类层，以便可以在新的数据集上进行训练。

我似乎无法在线找到有关使用SavedModels进行此操作的文档。

非常感谢您的帮助！