diff options
Diffstat (limited to 'import_pb_to_tensorboard')
-rwxr-xr-x | import_pb_to_tensorboard | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/import_pb_to_tensorboard b/import_pb_to_tensorboard new file mode 100755 index 0000000..2b77b14 --- /dev/null +++ b/import_pb_to_tensorboard @@ -0,0 +1,8 @@ +#!/usr/bin/python +# -*- coding: utf-8 -*- +import re +import sys +from tensorflow.python.tools.import_pb_to_tensorboard import main +if __name__ == '__main__': + sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0]) + sys.exit(main()) |