diff --git a/src/finn/core/throughput_test.py b/src/finn/core/throughput_test.py index fed44ce759ac4e639cc4d5f44e3fb1dba4ba77d8..ced14a02b32079919a485395dbfd8f3d19c77703 100644 --- a/src/finn/core/throughput_test.py +++ b/src/finn/core/throughput_test.py @@ -60,7 +60,7 @@ def throughput_test(model): process_compile = subprocess.Popen(bash_command, stdout=subprocess.PIPE) process_compile.communicate() - cmd = "sshpass -p {} scp -P{} {}@{}:{}/{}/nw_runtime.txt {}".format( + cmd = "sshpass -p {} scp -P{} {}@{}:{}/{}/nw_metrics.txt {}".format( pynq_password, pynq_port, pynq_username, @@ -73,6 +73,7 @@ def throughput_test(model): process_compile = subprocess.Popen(bash_command, stdout=subprocess.PIPE) process_compile.communicate() - file = open("{}/nw_runtime.txt".format(deployment_dir), "r") - res = file.read() + with open("{}/nw_metrics.txt".format(deployment_dir), "r") as file: + res = eval(file.read()) + return res