tf.train.write_graph
tf.train.write_graph
tf.train.write_graph
write_graph( graph_or_graph_def, logdir, name, as_text=True )
Defined in tensorflow/python/framework/graph_io.py
.
See the guide: Training > Training Utilities
Writes a graph proto to a file.
The graph is written as a binary proto unless as_text
is True
.
v = tf.Variable(0, name='my_variable') sess = tf.Session() tf.train.write_graph(sess.graph_def, '/tmp/my-model', 'train.pbtxt')
or
v = tf.Variable(0, name='my_variable') sess = tf.Session() tf.train.write_graph(sess.graph, '/tmp/my-model', 'train.pbtxt')
Args:
-
graph_or_graph_def
: AGraph
or aGraphDef
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
: IfTrue
, writes the graph as an ASCII proto.
Returns:
The path of the output proto file.
© 2017 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/api_docs/python/tf/train/write_graph