diff --git a/tools/device.py b/tools/device.py index 48474c9cdc03f955ce74c6b94669a1b7f55d400a..679d09e8fed23c4976052ef9af8aa66504178c40 100644 --- a/tools/device.py +++ b/tools/device.py @@ -763,13 +763,13 @@ class DeviceWrapper: '--input_node=%s' % ','.join(input_nodes), '--output_node=%s' % ','.join(output_nodes), '--input_shape=%s' % ':'.join(input_shapes), - '--output_shapes=%s' % ':'.join(output_shapes), + '--output_shape=%s' % ':'.join(output_shapes), '--input_file=%s/%s' % (model_output_dir, input_file_name), '--model_data_file=%s/%s.data' % (mace_model_dir, model_tag), '--device=%s' % device_type, '--omp_num_threads=%s' % omp_num_threads, - '--cpu_addinity_policy=%s' % cpu_affinity_policy, + '--cpu_affinity_policy=%s' % cpu_affinity_policy, '--gpu_perf_hint=%s' % gpu_perf_hint, '--gpu_priority_hint=%s' % gpu_priority_hint, '--model_file=%s' % mace_model_path