update the shell scripts

egs/librispeech/asr/conf/train_ctc_debug.yaml

@@ -13,7 +13,6 @@ seed: 1
 report-accuracy: True
 report-accuracy: True
 
 
 arch: s2t_transformer_s
 arch: s2t_transformer_s
-arch: s2t_sate
 share-decoder-input-output-embed: True
 share-decoder-input-output-embed: True
 optimizer: adam
 optimizer: adam
 clip-norm: 10.0
 clip-norm: 10.0

egs/mustc/st/conf/train_local_attn.yaml

+encoder-attention-type: local
+hard-mask-window: 20
+gauss-mask-sigma: 0
+init-mask-weight: 0
\ No newline at end of file

egs/mustc/st/run.sh

@@ -32,14 +32,14 @@ src_lang=en
 tgt_lang=de
 tgt_lang=de
 lang=${src_lang}-${tgt_lang}
 lang=${src_lang}-${tgt_lang}
 
 
-dataset=mustc-v2
+dataset=mustc
 task=speech_to_text
 task=speech_to_text
 vocab_type=unigram
 vocab_type=unigram
 asr_vocab_size=5000
 asr_vocab_size=5000
 vocab_size=10000
 vocab_size=10000
 share_dict=1
 share_dict=1
 speed_perturb=0
 speed_perturb=0
-lcrm=1
+lcrm=0
 tokenizer=0
 tokenizer=0
 
 
 use_specific_dict=0
 use_specific_dict=0
@@ -53,7 +53,7 @@ data_dir=~/st/data/${dataset}/st
 test_subset=tst-COMMON
 test_subset=tst-COMMON
 
 
 # exp
 # exp
-exp_prefix=${time}
+exp_prefix=$(date "+%m%d")
 extra_tag=
 extra_tag=
 extra_parameter=
 extra_parameter=
 exp_tag=baseline
 exp_tag=baseline
@@ -61,6 +61,8 @@ exp_name=
 
 
 # config
 # config
 train_config=train_ctc.yaml
 train_config=train_ctc.yaml
+train_config1=
+train_config2=
 
 
 # training setting
 # training setting
 fp16=1
 fp16=1
@@ -225,7 +227,15 @@ if [ ${stage} -le 1 ] && [ ${stop_stage} -ge 1 ]; then
         --save-dir ${model_dir}
         --save-dir ${model_dir}
         --tensorboard-logdir ${model_dir}"
         --tensorboard-logdir ${model_dir}"
 
 
-    if [[ -n ${extra_parameter} ]]; then
+    if [[ -n ${train_config1} ]]; then
+        cmd="${cmd}
+        --train-config1 ${train_config1}"
+    fi
+	if [[ -n ${train_config2} ]]; then
+        cmd="${cmd}
+        --train-config2 ${train_config2}"
+    fi
+	if [[ -n ${extra_parameter} ]]; then
         cmd="${cmd}
         cmd="${cmd}
         ${extra_parameter}"
         ${extra_parameter}"
     fi
     fi

egs/mustc/st/train.sh

@@ -3,8 +3,8 @@
 # training the model
 # training the model
 
 
 gpu_num=8
 gpu_num=8
-update_freq=2
-max_tokens=20000
+update_freq=1
+max_tokens=40000
 
 
 exp_name=
 exp_name=
 extra_tag=
 extra_tag=
@@ -28,6 +28,8 @@ train_config=train_ctc.yaml
 #train_config=train_ctc_sate_conformer.yaml
 #train_config=train_ctc_sate_conformer.yaml
 #train_config=train_ctc_sate_conformer_rpr.yaml
 #train_config=train_ctc_sate_conformer_rpr.yaml
 
 
+train_config1=train_local_attn.yaml
+
 cmd="./run.sh
 cmd="./run.sh
     --stage 1
     --stage 1
     --stop_stage 1
     --stop_stage 1
@@ -37,6 +39,9 @@ cmd="./run.sh
     --max_tokens ${max_tokens}
     --max_tokens ${max_tokens}
     "
     "
 
 
+if [[ -n ${train_config1} ]]; then
+    cmd="$cmd --train_config1 ${train_config1}"
+fi
 if [[ -n ${exp_name} ]]; then
 if [[ -n ${exp_name} ]]; then
     cmd="$cmd --exp_name ${exp_name}"
     cmd="$cmd --exp_name ${exp_name}"
 fi
 fi