[FastPitch/PyT] Fixed ckpt handling
This commit is contained in:
parent
2749a80b3a
commit
ff86473b48
|
@ -4,7 +4,7 @@ set -e
|
||||||
|
|
||||||
: ${MODEL_DIR:="pretrained_models/fastpitch"}
|
: ${MODEL_DIR:="pretrained_models/fastpitch"}
|
||||||
MODEL_ZIP="nvidia_fastpitch_200518.zip"
|
MODEL_ZIP="nvidia_fastpitch_200518.zip"
|
||||||
MODEL_CH="nvidia_fastpitch_200518.pt"
|
MODEL="nvidia_fastpitch_200518.pt"
|
||||||
MODEL_URL="https://api.ngc.nvidia.com/v2/models/nvidia/fastpitch_pyt_amp_ckpt_v1/versions/20.02.0/zip"
|
MODEL_URL="https://api.ngc.nvidia.com/v2/models/nvidia/fastpitch_pyt_amp_ckpt_v1/versions/20.02.0/zip"
|
||||||
|
|
||||||
mkdir -p "$MODEL_DIR"
|
mkdir -p "$MODEL_DIR"
|
||||||
|
@ -12,10 +12,16 @@ mkdir -p "$MODEL_DIR"
|
||||||
if [ ! -f "${MODEL_DIR}/${MODEL_ZIP}" ]; then
|
if [ ! -f "${MODEL_DIR}/${MODEL_ZIP}" ]; then
|
||||||
echo "Downloading ${MODEL_ZIP} ..."
|
echo "Downloading ${MODEL_ZIP} ..."
|
||||||
wget -qO ${MODEL_DIR}/${MODEL_ZIP} ${MODEL_URL} \
|
wget -qO ${MODEL_DIR}/${MODEL_ZIP} ${MODEL_URL} \
|
||||||
|| echo "ERROR: Failed to download ${MODEL_ZIP} from NGC" && exit 1
|
|| { echo "ERROR: Failed to download ${MODEL_ZIP} from NGC"; exit 1; }
|
||||||
fi
|
fi
|
||||||
|
|
||||||
if [ ! -f "${MODEL_DIR}/${MODEL_CH}" ]; then
|
if [ ! -f "${MODEL_DIR}/${MODEL}" ]; then
|
||||||
echo "Extracting ${MODEL_CH} ..."
|
echo "Extracting ${MODEL} ..."
|
||||||
unzip -qo ${MODEL_DIR}/${MODEL_ZIP} -d ${MODEL_DIR}
|
unzip -qo ${MODEL_DIR}/${MODEL_ZIP} -d ${MODEL_DIR} \
|
||||||
|
|| { echo "ERROR: Failed to extract ${MODEL_ZIP}"; exit 1; }
|
||||||
|
|
||||||
|
echo "OK"
|
||||||
|
|
||||||
|
else
|
||||||
|
echo "${MODEL} already downloaded."
|
||||||
fi
|
fi
|
||||||
|
|
|
@ -11,9 +11,9 @@ mkdir -p "$MODEL_DIR"
|
||||||
if [ ! -f "${MODEL_DIR}/${MODEL}" ]; then
|
if [ ! -f "${MODEL_DIR}/${MODEL}" ]; then
|
||||||
echo "Downloading ${MODEL} ..."
|
echo "Downloading ${MODEL} ..."
|
||||||
wget --content-disposition -qO ${MODEL_DIR}/${MODEL} ${MODEL_URL} \
|
wget --content-disposition -qO ${MODEL_DIR}/${MODEL} ${MODEL_URL} \
|
||||||
|| echo "ERROR: Failed to download ${MODEL} from NGC" && exit 1
|
|| { echo "ERROR: Failed to download ${MODEL} from NGC"; exit 1; }
|
||||||
echo "OK"
|
echo "OK"
|
||||||
|
|
||||||
else
|
else
|
||||||
echo "${MODEL}.pt already downloaded."
|
echo "${MODEL} already downloaded."
|
||||||
fi
|
fi
|
||||||
|
|
|
@ -3,7 +3,7 @@
|
||||||
set -e
|
set -e
|
||||||
|
|
||||||
: ${MODEL_DIR:="pretrained_models/waveglow"}
|
: ${MODEL_DIR:="pretrained_models/waveglow"}
|
||||||
MODEL="nvidia_waveglow256pyt_fp16"
|
MODEL="nvidia_waveglow256pyt_fp16.pt"
|
||||||
MODEL_ZIP="waveglow_ckpt_amp_256_20.01.0.zip"
|
MODEL_ZIP="waveglow_ckpt_amp_256_20.01.0.zip"
|
||||||
MODEL_URL="https://api.ngc.nvidia.com/v2/models/nvidia/waveglow_ckpt_amp_256/versions/20.01.0/zip"
|
MODEL_URL="https://api.ngc.nvidia.com/v2/models/nvidia/waveglow_ckpt_amp_256/versions/20.01.0/zip"
|
||||||
|
|
||||||
|
@ -12,16 +12,16 @@ mkdir -p "$MODEL_DIR"
|
||||||
if [ ! -f "${MODEL_DIR}/${MODEL_ZIP}" ]; then
|
if [ ! -f "${MODEL_DIR}/${MODEL_ZIP}" ]; then
|
||||||
echo "Downloading ${MODEL_ZIP} ..."
|
echo "Downloading ${MODEL_ZIP} ..."
|
||||||
wget --content-disposition -qO ${MODEL_DIR}/${MODEL_ZIP} ${MODEL_URL} \
|
wget --content-disposition -qO ${MODEL_DIR}/${MODEL_ZIP} ${MODEL_URL} \
|
||||||
|| echo "ERROR: Failed to download ${MODEL_ZIP} from NGC" && exit 1
|
|| { echo "ERROR: Failed to download ${MODEL_ZIP} from NGC"; exit 1; }
|
||||||
fi
|
fi
|
||||||
|
|
||||||
if [ ! -f "${MODEL_DIR}/${MODEL}.pt" ]; then
|
if [ ! -f "${MODEL_DIR}/${MODEL}" ]; then
|
||||||
echo "Extracting ${MODEL} ..."
|
echo "Extracting ${MODEL} ..."
|
||||||
unzip -qo ${MODEL_DIR}/${MODEL_ZIP} -d ${MODEL_DIR} \
|
unzip -qo ${MODEL_DIR}/${MODEL_ZIP} -d ${MODEL_DIR} \
|
||||||
|| echo "ERROR: Failed to extract ${MODEL_ZIP}" && exit 1
|
|| { echo "ERROR: Failed to extract ${MODEL_ZIP}"; exit 1; }
|
||||||
|
|
||||||
echo "OK"
|
echo "OK"
|
||||||
|
|
||||||
else
|
else
|
||||||
echo "${MODEL}.pt already downloaded."
|
echo "${MODEL} already downloaded."
|
||||||
fi
|
fi
|
||||||
|
|
|
@ -10,7 +10,7 @@
|
||||||
|
|
||||||
[ "$AMP" = true ] && AMP_FLAG="--amp"
|
[ "$AMP" = true ] && AMP_FLAG="--amp"
|
||||||
|
|
||||||
mkdir -o "$OUTPUT_DIR"
|
mkdir -p "$OUTPUT_DIR"
|
||||||
|
|
||||||
for BS in $BS_SEQUENCE ; do
|
for BS in $BS_SEQUENCE ; do
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue