如何保存和还原使用Tensorflow Hub模块的图形

Question

我正在寻找一个使用TF集线器映像模块的工作示例，针对自定义任务对其进行微调，并向后传播到集线器模块。然后保存此结果模型并重新加载以进行评估。

我看过https://github.com/tensorflow/hub/blob/master/examples/image_retraining/retrain.py，但我认为它不会训练模块。

Answer 1

从检查点还原模型时，它照常工作。

• 如果要从检查点还原模型，则模块变量也将从该模型还原（而不是从原始模块加载）。

• 将模块变量设置为原始值的唯一时间是它们的初始化程序运行时。