fix the bug of training mdtc_bs16_fp32

solve the bug of training mdtc_bs16_fp32
pull/2354/head
Zhao Yuting 2 years ago committed by GitHub
parent 43af22e4e6
commit 975ceb811a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -22,7 +22,7 @@ model_name=$(func_parser_value "${lines[1]}")
echo "model_name:"${model_name}
trainer_list=$(func_parser_value "${lines[14]}")
if [ ${MODE} = "benchmark_train" ];then
if [[ ${MODE} = "benchmark_train" ]];then
curPath=$(readlink -f "$(dirname "$0")")
echo "curPath:"${curPath} # /PaddleSpeech/tests/test_tipc
cd ${curPath}/../..
@ -36,10 +36,10 @@ if [ ${MODE} = "benchmark_train" ];then
pip install jsonlines
pip list
cd -
if [ ${model_name} == "conformer" ]; then
if [[ ${model_name} == "conformer" ]]; then
# set the URL for aishell_tiny dataset
conformer_aishell_URL=${conformer_aishell_URL:-"None"}
if [ ${conformer_aishell_URL} == 'None' ];then
if [[ ${conformer_aishell_URL} == 'None' ]];then
echo "please contact author to get the URL.\n"
exit
else
@ -66,7 +66,7 @@ if [ ${MODE} = "benchmark_train" ];then
sed -i "s#data/#test_tipc/conformer/benchmark_train/data/#g" ${curPath}/conformer/benchmark_train/conf/preprocess.yaml
fi
if [ ${model_name} == "pwgan" ]; then
if [[ ${model_name} == "pwgan" ]]; then
# 下载 csmsc 数据集并解压缩
wget -nc https://weixinxcxdb.oss-cn-beijing.aliyuncs.com/gwYinPinKu/BZNSYP.rar
mkdir -p BZNSYP
@ -80,9 +80,10 @@ if [ ${MODE} = "benchmark_train" ];then
python ../paddlespeech/t2s/exps/gan_vocoder/normalize.py --metadata=dump/test/raw/metadata.jsonl --dumpdir=dump/test/norm --stats=dump/train/feats_stats.npy
fi
if [ ${model_name} == "mdtc" ]; then
if [[ ${model_name} == "mdtc" ]]; then
# 下载 Snips 数据集并解压缩
wget -nc https://paddlespeech.bj.bcebos.com/datasets/hey_snips_kws_4.0.tar.gz.1 https://paddlespeech.bj.bcebos.com/datasets/hey_snips_https://paddlespeech.bj.bcebos.com/datasets/hey_snips_kws_4.0.tar.gz.2
wget https://paddlespeech.bj.bcebos.com/datasets/hey_snips_kws_4.0.tar.gz.1
wget https://paddlespeech.bj.bcebos.com/datasets/hey_snips_kws_4.0.tar.gz.2
cat hey_snips_kws_4.0.tar.gz.* > hey_snips_kws_4.0.tar.gz
rm hey_snips_kws_4.0.tar.gz.*
tar -xzf hey_snips_kws_4.0.tar.gz

Loading…
Cancel
Save