experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

docs: update docs for tf.python.tools.import_pb_to_tensorboard

Browse Source
This commit is contained in:
Xiaoquan Kong 2020-03-12 16:26:47 +08:00
parent 473d139b18
commit bce90b6e09
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/python/tools/import_pb_to_tensorboard.py
View File

@ -41,19 +41,19 @@ except ImportError:
def import_to_tensorboard(model_dir, log_dir, tag_set):
"""View an imported protobuf model (`.pb` file) as a graph in Tensorboard.
"""View an SavedModel as a graph in Tensorboard.
Args:
model_dir: The location of the protobuf (`pb`) model to visualize
model_dir: The directory containing the SavedModel to import.
log_dir: The location for the Tensorboard log to begin visualization from.
tag_set: Group of tag(s) of the MetaGraphDef to load, in string format,
separated by ','. For tag-set contains multiple tags, all tags must be
passed in.
Usage:
Call this function with your model location and desired log directory.
Call this function with your SavedModel location and desired log directory.
Launch Tensorboard by pointing it to the log directory.
View your imported `.pb` model as a graph.
View your imported SavedModel as a graph.
"""
with session.Session(graph=ops.Graph()) as sess:
input_graph_def = saved_model_utils.get_meta_graph_def(model_dir,