diff --git a/hpvm/scripts/download_weights.sh b/hpvm/scripts/download_weights.sh new file mode 100755 index 0000000000000000000000000000000000000000..757abbf3b6f442e729fc100dad73605511e0454f --- /dev/null +++ b/hpvm/scripts/download_weights.sh @@ -0,0 +1,10 @@ +#!/bin/bash +# 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 +# Download the zip file from google drive +./gdown.pl 'https://drive.google.com/file/d/1V_yd9sKcZQ7zhnO5YhRpOsaBPLEEvM9u' model_params.zip +unzip model_params.zip # should give a "model_params" folder +# All our benchmarks also know to look for parameters in <build_dir>/model_params. +# Cleanup: +rm gdown.pl model_params.zip \ No newline at end of file diff --git a/hpvm/scripts/llvm_installer.sh b/hpvm/scripts/llvm_installer.sh index 21ed6ee6d13ef83e0cc62f643d8e674e7c0e5a90..0cbaea8e493de8a08833ca5ab025e2692f609fe5 100755 --- a/hpvm/scripts/llvm_installer.sh +++ b/hpvm/scripts/llvm_installer.sh @@ -256,17 +256,7 @@ 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 - # Download the zip file from google drive - ./gdown.pl 'https://drive.google.com/file/d/1V_yd9sKcZQ7zhnO5YhRpOsaBPLEEvM9u' model_params.zip - unzip model_params.zip # should give a "model_params" folder - mv model_params $BUILD_DIR - # All our benchmarks also know to look for parameters in <build_dir>/model_params. - # Cleanup: - rm gdown.pl model_params.zip + ../scripts/download_weights.sh else echo "Skipping weight download" fi