diff options
| author | alexander <alexander.efremov@arm.com> | 2021-05-12 19:06:02 +0100 |
|---|---|---|
| committer | Kshitij Sisodia <kshitij-sisodia-arm@review.mlplatform.org> | 2021-05-18 08:31:49 +0000 |
| commit | 50a0650cd3c804a89aa77815820a8df390245c2f (patch) | |
| tree | 564d00dfe6db460926c36ef0309708304870d1c5 /set_up_default_resources.py | |
| parent | ddcb56d603ffa9cbff173d3c7066c2fa6f8d3c4c (diff) | |
| download | ml-embedded-evaluation-kit-50a0650cd3c804a89aa77815820a8df390245c2f.tar.gz | |
MLECO-1873: default models are for 128 MAC config
Signed-off-by: alexander <alexander.efremov@arm.com>
Change-Id: Ic3e940a9d289d649976c4e951ced1473fc31ff3f
Diffstat (limited to 'set_up_default_resources.py')
| -rw-r--r-- | set_up_default_resources.py | 20 |
1 files changed, 15 insertions, 5 deletions
diff --git a/set_up_default_resources.py b/set_up_default_resources.py index 60c1747..0a02395 100644 --- a/set_up_default_resources.py +++ b/set_up_default_resources.py @@ -93,10 +93,14 @@ def call_command(command: str) -> str: command (string): Specifies the command to run. """ logging.info(command) - proc = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, shell=True) - stdout_log = proc.communicate()[0].decode("utf-8") - logging.info(stdout_log) - return stdout_log + proc = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, shell=True) + log = proc.stdout.decode("utf-8") + if proc.returncode == 0: + logging.info(log) + else: + logging.error(log) + proc.check_returncode() + return log def set_up_resources(run_vela_on_models=False): @@ -185,7 +189,7 @@ def set_up_resources(run_vela_on_models=False): # New models will have same name with '_vela' appended. # For example: # original model: ds_cnn_clustered_int8.tflite - # after vela model: ds_cnn_clustered_int8_vela.tflite + # after vela model: ds_cnn_clustered_int8_vela_H128.tflite # # Note: To avoid to run vela twice on the same model, it's supposed that # downloaded model names don't contain the 'vela' word. @@ -205,6 +209,12 @@ def set_up_resources(run_vela_on_models=False): "--system-config=Ethos_U55_High_End_Embedded " + f"--output-dir={output_dir}") call_command(command) + # model name after compiling with vela is an initial model name + _vela suffix + vela_optimised_model_path = str(model).replace(".tflite", "_vela.tflite") + # we want it to be initial model name + _vela_H128 suffix which indicates selected MAC config. + new_vela_optimised_model_path = vela_optimised_model_path.replace("_vela.tflite", "_vela_H128.tflite") + # rename default vela model + os.rename(vela_optimised_model_path, new_vela_optimised_model_path) if __name__ == '__main__': |