Skip to content
Snippets Groups Projects
Commit 6c72a5e7 authored by Akash Kothari's avatar Akash Kothari
Browse files

Move the message of dowloading weights up in the install script

parent 464bbf2b
No related branches found
No related tags found
No related merge requests found
......@@ -32,6 +32,7 @@ HPVM_RT=hpvm-rt/hpvm-rt.bc
TARGET=all
TARGET_INPUT=all
FLAGGED=false
DOWNLOAD_WEIGHTS=false
# Get flags
while getopts 'hmj:t:' opt; do
......@@ -83,6 +84,7 @@ if $FLAGGED; then
echo Automated: $AUTOMATE
echo Threads: $NUM_THREADS
echo Targets: $TARGET
echo Download Weights: $DOWNLOAD_WEIGHTS
echo
else
echo "No Flags found. Using command line prompts."
......@@ -122,11 +124,22 @@ else
TARGET=$TARGET_INPUT
fi
echo
read_yn "Download weights necessary to run DNN benchmarks?" LOAD_WEIGHTS
if [[ $LOAD_WEIGHTS == "" ]]; then
echo "No input given. Weights will not be downloaded."
elif [[ $LOAD_WEIGHTS == "n" ]]; then
echo "Weights will not be downloaded."
else
DOWNLOAD_WEIGHTS=$LOAD_WEIGHTS
fi
echo
echo "Running with the following options:"
echo Automated: $AUTOMATE
echo Threads: $NUM_THREADS
echo Targets: $TARGET
echo Download Weights: $DOWNLOAD_WEIGHTS
echo
fi
......@@ -239,8 +252,11 @@ echo make -j$NUM_THREADS
make -j$NUM_THREADS
#make install
read_yn "Download weights necessary to run DNN benchmarks?" DOWNLOAD_WEIGHTS
if [ $DOWNLOAD_WEIGHTS == "y" ]; then
echo
echo "Downloading weights for DNN benchmarks..."
echo
# First get hands on gdown -- google drive downloader
wget https://raw.githubusercontent.com/circulosmeos/gdown.pl/master/gdown.pl -O gdown.pl
chmod +x ./gdown.pl
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment