diff --git a/egs/librispeech/ASR/.run_v3.sh.swp b/egs/librispeech/ASR/.run_v3.sh.swp index b52981109..18bd96fe9 100644 Binary files a/egs/librispeech/ASR/.run_v3.sh.swp and b/egs/librispeech/ASR/.run_v3.sh.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp index 3a7e87bde..d08b0fb6e 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py index f7289b5ab..033954af5 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py @@ -1231,6 +1231,7 @@ def run(rank, world_size, args, wb=None): if rank == 0: print(enc_names) + optimizer_enc = ScaledAdam( enc_param, lr=params.peak_enc_lr, diff --git a/egs/librispeech/ASR/run_v3.sh b/egs/librispeech/ASR/run_v3.sh index 6db23ca38..b2cbc95bd 100755 --- a/egs/librispeech/ASR/run_v3.sh +++ b/egs/librispeech/ASR/run_v3.sh @@ -13,7 +13,7 @@ export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7" --full-libri 0 \ --exp-dir ./pruned_transducer_stateless_d2v_v2/$1 \ --max-duration 250 \ - --freeze-finetune-updates 500 \ + --freeze-finetune-updates 200 \ --use-fp16 1 \ --peak-enc-lr 0.0001 \ --peak-dec-lr 0.003 \