diff --git a/tests/test_tipc/configs/mdtc/train_infer_python.txt b/tests/test_tipc/configs/mdtc/train_infer_python.txt new file mode 100644 index 00000000..7a5f658e --- /dev/null +++ b/tests/test_tipc/configs/mdtc/train_infer_python.txt @@ -0,0 +1,57 @@ +===========================train_params=========================== +model_name:mdtc +python:python3.7 +gpu_list:0|0,1 +null:null +null:null +--benchmark-max-step:50 +null:null +--benchmark-batch-size:16 +null:null +null:null +null:null +null:null +## +trainer:norm_train +norm_train: ../paddlespeech/kws/exps/mdtc/train.py --config=../examples/hey_snips/kws0/conf/mdtc.yaml +pact_train:null +fpgm_train:null +distill_train:null +null:null +null:null +## +===========================eval_params=========================== +eval:null +null:null +## +===========================infer_params=========================== +null:null +null:null +norm_export: null +quant_export:null +fpgm_export:null +distill_export:null +export1:null +export2:null +null:null +infer_model:null +infer_export:null +infer_quant:null +inference:null +null:null +null:null +null:null +null:null +null:null +null:null +null:null +null:null +null:null +null:null +null:null +===========================train_benchmark_params========================== +batch_size:16|30 +fp_items:fp32 +iteration:50 +--profiler-options:"batch_range=[10,35];state=GPU;tracer_option=Default;profile_path=model.profile" +flags:null diff --git a/tests/test_tipc/prepare.sh b/tests/test_tipc/prepare.sh index a1393801..b38bbcba 100644 --- a/tests/test_tipc/prepare.sh +++ b/tests/test_tipc/prepare.sh @@ -80,4 +80,13 @@ 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 + # 下载 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 + 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 + # 解压后的数据目录 ./hey_snips_research_6k_en_train_eval_clean_ter + fi + fi