如何保存和恢复使用Tensorflow Hub模块的图形
我正在寻找一个工作示例,其中我使用TF Hub映像模块,为自定义任务对其进行微调,并将其反向传播到Hub模块。然后保存此结果模型并重新加载以进行评估如何保存和恢复使用Tensorflow Hub模块的图形,tensorflow,tensorflow-hub,Tensorflow,Tensorflow Hub,我正在寻找一个工作示例,其中我使用TF Hub映像模块,为自定义任务对其进行微调,并将其反向传播到Hub模块。然后保存此结果模型并重新加载以进行评估 我已经看过了,但我认为它没有训练模块。它只是在从检查点恢复模型时正常工作 如果您的模型是从检查点恢复的,那么模块变量也将从检查点恢复(而不是从原始模块加载) 模块变量设置为原始值的唯一时间是在其初始值设定项运行时 您是否设法使其正常工作?如果是这样,怎么做?@cipriantomoagăI能够让它工作。然而,我认为自从我发布我的问题以来,这个文档
我已经看过了,但我认为它没有训练模块。它只是在从检查点恢复模型时正常工作
- 如果您的模型是从检查点恢复的,那么模块变量也将从检查点恢复(而不是从原始模块加载)
- 模块变量设置为原始值的唯一时间是在其初始值设定项运行时