diff --git a/src/finn/core/remote_exec.py b/src/finn/core/remote_exec.py index fbc3cfe84c37d1deec3585929470936d54ce9106..5c330e29a52ab73343fbf7fd1858020b3d0cdd30 100644 --- a/src/finn/core/remote_exec.py +++ b/src/finn/core/remote_exec.py @@ -64,7 +64,8 @@ def remote_exec(model, execution_context): process_compile.communicate() cmd = ( "sshpass -p {} ssh {}@{} -p {} " - '"cd {}/{}; echo "{}" | sudo -S python3.6 driver.py remote_pynq"' + '"cd {}/{}; echo "{}" | ' + 'sudo -S python3.6 driver.py remote_pynq 1 resizer.bit input.npy output.npy"' ).format( pynq_password, pynq_username, diff --git a/src/finn/core/throughput_test.py b/src/finn/core/throughput_test.py index ced14a02b32079919a485395dbfd8f3d19c77703..dda30b638255562e0dd1795ae5dc7586177308dc 100644 --- a/src/finn/core/throughput_test.py +++ b/src/finn/core/throughput_test.py @@ -46,7 +46,9 @@ def throughput_test(model): cmd = ( "sshpass -p {} ssh {}@{} -p {} " - '"cd {}/{}; echo "{}" | sudo -S python3.6 driver.py throughput_test"' + '"cd {}/{}; echo "{}" | ' + "sudo -S python3.6 driver.py throughput_test 1000 " + 'resizer.bit input.npy output.npy"' ).format( pynq_password, pynq_username,