在張量流中使用tf.save時的模型格式

Question

正如我們所知，我們可以在訓練時使用tf.save()保存模型。

但是，存儲的模型文件的格式是什麼？是否可以使用fread或其他什麼來讀取文件？

Answer 1

您可以從https://www.tensorflow.org/versions/r0.8/how_tos/tool_developers/index.html找到有關該文件格式以及如何解析的詳細信息。

見一個簡單的例子：

graph_def = graph_pb2.GraphDef() 
with open(FLAGS.graph, "rb") as f: 
    if FLAGS.input_binary: 
    graph_def.ParseFromString(f.read) 
    else: 
    text_format.Merge(f.read(), graph_def) 

你可以找到在網頁中更多的例子。