tf.io.write_graph
Writes a graph proto to a file.
tf.io.write_graph(
graph_or_graph_def, logdir, name, as_text=True
)
The graph is written as a text proto unless as_text is False.
v = tf.Variable(0, name='my_variable')
sess = tf.compat.v1.Session()
tf.io.write_graph(sess.graph_def, '/tmp/my-model', 'train.pbtxt')
or
v = tf.Variable(0, name='my_variable')
sess = tf.compat.v1.Session()
tf.io.write_graph(sess.graph, '/tmp/my-model', 'train.pbtxt')
| Args | |
|---|---|
graph_or_graph_def |
A Graph or a GraphDef protocol buffer. |
logdir |
Directory where to write the graph. This can refer to remote filesystems, such as Google Cloud Storage (GCS). |
name |
Filename for the graph. |
as_text |
If True, writes the graph as an ASCII proto. |
| Returns | |
|---|---|
| The path of the output proto file. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/io/write_graph