aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/tools
diff options
context:
space:
mode:
authorGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-07 12:30:34 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-07 12:30:34 -0700
commitbb096a0735445c7e05be4b68042f21660602001c (patch)
tree427041f38531fbdb8dd7a21d0ac6ef86e15f202b /tensorflow/python/tools
parenta65d3dd42122d3a58985d56118d58c5b4224f38f (diff)
parent94ef0a70717d83316042cba924e70fe024a51661 (diff)
Merge pull request #21768 from superbobry:fix-load_inputs_from_input_arg_string
PiperOrigin-RevId: 212017232
Diffstat (limited to 'tensorflow/python/tools')
-rw-r--r--tensorflow/python/tools/saved_model_cli.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/tools/saved_model_cli.py b/tensorflow/python/tools/saved_model_cli.py
index 6716c79f87..c5289564fe 100644
--- a/tensorflow/python/tools/saved_model_cli.py
+++ b/tensorflow/python/tools/saved_model_cli.py
@@ -546,7 +546,7 @@ def load_inputs_from_input_arg_string(inputs_str, input_exprs_str,
input_examples = preprocess_input_examples_arg_string(input_examples_str)
for input_tensor_key, (filename, variable_name) in inputs.items():
- data = np.load(file_io.FileIO(filename, mode='r'))
+ data = np.load(file_io.FileIO(filename, mode='rb'))
# When a variable_name key is specified for the input file
if variable_name: