From 560ef5f40cecd94d68194f8f602cbb42d75e80dd Mon Sep 17 00:00:00 2001 From: Bin Li Date: Wed, 26 Dec 2018 19:11:23 +0800 Subject: [PATCH] Fix input_dir/output_dir for example --- tools/device.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tools/device.py b/tools/device.py index f0a91a56..298b997f 100644 --- a/tools/device.py +++ b/tools/device.py @@ -173,6 +173,8 @@ class DeviceWrapper: gpu_priority_hint=3, input_file_name='model_input', output_file_name='model_out', + input_dir="", + output_dir="", runtime_failure_ratio=0.0, address_sanitizer=False, link_dynamic=False, @@ -208,6 +210,8 @@ class DeviceWrapper: input_file_name), "--output_file=%s/%s" % (model_output_dir, output_file_name), + "--input_dir=%s" % input_dir, + "--output_dir=%s" % output_dir, "--model_data_file=%s/%s.data" % (mace_model_dir, model_tag), "--device=%s" % device_type, @@ -296,6 +300,8 @@ class DeviceWrapper: "--output_shape=%s" % ":".join(output_shapes), "--input_file=%s/%s" % (self.data_dir, input_file_name), "--output_file=%s/%s" % (self.data_dir, output_file_name), + "--input_dir=%s" % input_dir, + "--output_dir=%s" % output_dir, "--model_data_file=%s/%s.data" % (self.data_dir, model_tag), "--device=%s" % device_type, "--round=%s" % running_round, @@ -553,6 +559,8 @@ class DeviceWrapper: libmace_dynamic_library_path=LIBMACE_DYNAMIC_PATH, link_dynamic=link_dynamic, quantize_stat=flags.quantize_stat, + input_dir=flags.input_dir, + output_dir=flags.output_dir, ) if flags.validate: model_file_path, weight_file_path = get_model_files( -- GitLab