From fedaf166f6bf27fff7917f604a83152801f1934a Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Thu, 25 Mar 2021 16:21:12 -0500
Subject: [PATCH] Fixed interactive arguments of hpvm installer
---
hpvm/scripts/hpvm_installer.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/hpvm/scripts/hpvm_installer.py b/hpvm/scripts/hpvm_installer.py
index e83d5f3e72..f99a333a60 100755
--- a/hpvm/scripts/hpvm_installer.py
+++ b/hpvm/scripts/hpvm_installer.py
@@ -120,11 +120,15 @@ def prompt_args():
"Build target: ", parse_targets, "Input shouldn't contain space"
)
+ download_weights = input_with_check(
+ "Download DNN weights (recommended)? [y/n]: ", parse_yn, "Please enter y or n"
+ )
run_tests = input_with_check(
"Build and run tests? [y/n]: ", parse_yn, "Please enter y or n"
)
return Namespace(
- no_build=not auto_build, parallel=threads, targets=targets, run_tests=run_tests
+ no_build=not auto_build, parallel=threads, targets=targets, run_tests=run_tests,
+ no_params=not download_weights
)
--
GitLab