如何从迁移学习训练中冻结图形

时间:2020-04-29 15:24:51

标签: tensorflow transfer-learning

我使用转移学习方法训练模型: https://www.tensorflow.org/tutorials/images/transfer_learning

在火车结束时,我使用rhis创建一个pb文件:

 # Save the entire model as a SavedModel.
    model.save(os.path.join(FLAGS.output_result,'saved_model/my_model'))

我想将该pb文件用于该项目: https://github.com/tensorflow/tensorflow/tree/master/tensorflow/examples/label_image

当我要加载pb文件时,出现该错误:

 Error Can't open graph: Not found: Failed to load compute graph at

文件路径已正确设置。我认为这来自pb文件本身。

0 个答案:

没有答案