diff options
author | alexander <alexander.efremov@arm.com> | 2021-05-24 18:56:32 +0100 |
---|---|---|
committer | Kshitij Sisodia <kshitij.sisodia@arm.com> | 2021-05-25 15:53:28 +0100 |
commit | 3ef1fd4880d76f62ee25c8a6f2afaa3f869498c1 (patch) | |
tree | 15ae35f1c7fcdf15e8063c1a9d57a969e8dc74e1 /set_up_default_resources.py | |
parent | 3c8256df7de49a4fb64cdbcdea46ff471ff5f846 (diff) | |
download | ml-embedded-evaluation-kit-3ef1fd4880d76f62ee25c8a6f2afaa3f869498c1.tar.gz |
MLECO-1944: Minor documentation clean-up done and script to download TPIP added21.05-rc1
Signed-off-by: alexander <alexander.efremov@arm.com>
Change-Id: Id4a9b220ce753a5ecf3483e86d837c1814dc7fb9
Diffstat (limited to 'set_up_default_resources.py')
-rwxr-xr-x | set_up_default_resources.py | 34 |
1 files changed, 22 insertions, 12 deletions
diff --git a/set_up_default_resources.py b/set_up_default_resources.py index 7639364..362552a 100755 --- a/set_up_default_resources.py +++ b/set_up_default_resources.py @@ -176,14 +176,18 @@ def set_up_resources(run_vela_on_models=False): res_dst = os.path.join(download_dir, uc["use_case_name"], res_name) - try: - g = urllib.request.urlopen(res_url) - with open(res_dst, 'b+w') as f: - f.write(g.read()) - logging.info(f"- Downloaded {res_url} to {res_dst}.") - except URLError: - logging.error(f"URLError while downloading {res_url}.") - raise + + if os.path.isfile(res_dst): + logging.info(f"File {res_dst} exists, skipping download.") + else: + try: + g = urllib.request.urlopen(res_url) + with open(res_dst, 'b+w') as f: + f.write(g.read()) + logging.info(f"- Downloaded {res_url} to {res_dst}.") + except URLError: + logging.error(f"URLError while downloading {res_url}.") + raise # 3. Run vela on models in resources_downloaded # New models will have same name with '_vela' appended. @@ -201,6 +205,15 @@ def set_up_resources(run_vela_on_models=False): for model in models: output_dir = os.path.dirname(model) + # 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") + + if os.path.isfile(new_vela_optimised_model_path): + logging.info(f"File {new_vela_optimised_model_path} exists, skipping optimisation.") + continue + command = (f". {env_activate} && vela {model} " + "--accelerator-config=ethos-u55-128 " + "--block-config-limit=0 " + @@ -209,10 +222,7 @@ 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) logging.info(f"Renaming {vela_optimised_model_path} to {new_vela_optimised_model_path}.") |