代码之家  ›  专栏  ›  技术社区  ›  dgumo

如何保存和恢复使用Tensorflow Hub模块的图形

  •  0
  • dgumo  · 技术社区  · 8 年前

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

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

    1 回复  |  直到 8 年前
        1
  •  0
  •   André Susano Pinto    8 年前

    当从检查点恢复模型时,它会像往常一样工作。

    • 模块变量设置为原始值的唯一时间是在其初始值设定项运行时。

    推荐文章