From 3d5721023200a7c81d57cfa4c714df670683b47f Mon Sep 17 00:00:00 2001 From: auphelia <jakobapk@web.de> Date: Tue, 5 May 2020 16:39:23 +0100 Subject: [PATCH] [Core] Change ssh driver.py commands to fit new argument style --- src/finn/core/remote_exec.py | 2 +- src/finn/core/throughput_test.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/src/finn/core/remote_exec.py b/src/finn/core/remote_exec.py index e97eb19a1..afea8f74f 100644 --- a/src/finn/core/remote_exec.py +++ b/src/finn/core/remote_exec.py @@ -65,7 +65,7 @@ def remote_exec(model, execution_context): cmd = ( "sshpass -p {} ssh {}@{} -p {} " '"cd {}/{}; echo "{}" | ' - 'sudo -S python3.6 driver.py remote_pynq 1 resizer.bit input.npy output.npy"' + 'sudo -S python3.6 driver.py --exec_mode="remote_pynq" --batchsize=1"' ).format( pynq_password, pynq_username, diff --git a/src/finn/core/throughput_test.py b/src/finn/core/throughput_test.py index fc929237b..c82d540e2 100644 --- a/src/finn/core/throughput_test.py +++ b/src/finn/core/throughput_test.py @@ -47,8 +47,7 @@ def throughput_test(model): cmd = ( "sshpass -p {} ssh {}@{} -p {} " '"cd {}/{}; echo "{}" | ' - "sudo -S python3.6 driver.py throughput_test 1000 " - 'resizer.bit input.npy output.npy"' + 'sudo -S python3.6 driver.py --exec_mode="throughput_test" --batchsize=1000"' ).format( pynq_password, pynq_username, -- GitLab