diff options
| author |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-07 12:30:34 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-07 12:30:34 -0700 |
| commit | bb096a0735445c7e05be4b68042f21660602001c (patch) | |
| tree | 427041f38531fbdb8dd7a21d0ac6ef86e15f202b /tensorflow/python/tools | |
| parent | a65d3dd42122d3a58985d56118d58c5b4224f38f (diff) | |
| parent | 94ef0a70717d83316042cba924e70fe024a51661 (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.py | 2 |
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: |