From bce90b6e090ed4743ec63e86a6e6321d6049e627 Mon Sep 17 00:00:00 2001 From: Xiaoquan Kong Date: Thu, 12 Mar 2020 16:26:47 +0800 Subject: [PATCH] docs: update docs for tf.python.tools.import_pb_to_tensorboard --- tensorflow/python/tools/import_pb_to_tensorboard.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tensorflow/python/tools/import_pb_to_tensorboard.py b/tensorflow/python/tools/import_pb_to_tensorboard.py index 5e9400d6d25..5cd5df7b231 100644 --- a/tensorflow/python/tools/import_pb_to_tensorboard.py +++ b/tensorflow/python/tools/import_pb_to_tensorboard.py @@ -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,