tensorflow - 如何保存和恢复使用 Tensorflow Hub 模块的图表

Question

我正在寻找一个工作示例，其中我使用 TF Hub 图像模块，为自定义任务微调它，也反向传播到 hub 模块。然后保存这个生成的模型并重新加载它以进行评估。

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

Answer 1

从检查点恢复模型时，它就像往常一样工作。

• 如果您的模型是从检查点恢复的，那么模块变量也将从检查点恢复（而不是从原始模块加载）。

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