如何保存和恢复使用 Tensorflow Hub 模块的图表
How can I save and restore a graph that uses Tensorflow Hub Module
我正在寻找一个工作示例，我在其中使用 TF Hub 图像模块，针对自定义任务对其进行微调，同时反向传播到 hub 模块。然后保存生成的模型并重新加载以进行评估。
我看过 https://github.com/tensorflow/hub/blob/master/examples/image_retraining/retrain.py，但我认为它无法训练模块。
从检查点恢复模型时它就像往常一样工作。
如果您的模型是从检查点恢复的，那么模块变量也将从它恢复（而不是从原始模块加载）。
唯一一次将模块变量设置为原始值是在它们的初始值设定项为 运行.
我正在寻找一个工作示例，我在其中使用 TF Hub 图像模块，针对自定义任务对其进行微调，同时反向传播到 hub 模块。然后保存生成的模型并重新加载以进行评估。
我看过 https://github.com/tensorflow/hub/blob/master/examples/image_retraining/retrain.py，但我认为它无法训练模块。
从检查点恢复模型时它就像往常一样工作。
如果您的模型是从检查点恢复的，那么模块变量也将从它恢复（而不是从原始模块加载）。
唯一一次将模块变量设置为原始值是在它们的初始值设定项为 运行.
时