diff options
Diffstat (limited to 'tensorflow/python/tools/import_pb_to_tensorboard.py')
-rw-r--r-- | tensorflow/python/tools/import_pb_to_tensorboard.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/tools/import_pb_to_tensorboard.py b/tensorflow/python/tools/import_pb_to_tensorboard.py index 2cab409cfb..a8712fc37e 100644 --- a/tensorflow/python/tools/import_pb_to_tensorboard.py +++ b/tensorflow/python/tools/import_pb_to_tensorboard.py @@ -64,11 +64,13 @@ if __name__ == "__main__": "--model_dir", type=str, default="", + required=True, help="The location of the protobuf (\'pb\') model to visualize.") parser.add_argument( "--log_dir", type=str, default="", + required=True, help="The location for the Tensorboard log to begin visualization from.") FLAGS, unparsed = parser.parse_known_args() app.run(main=main, argv=[sys.argv[0]] + unparsed) |