mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2022-12-13 13:58:47 +09:00
parent e039441a0f
commit 24ba63ed5b
4 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/.run_v3.sh.swp
View File

Binary file not shown.

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
View File

Binary file not shown.

1
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
View File

@ -1013,6 +1013,7 @@ def train_one_epoch(
scaler.scale(loss).backward() scaler.scale(loss).backward()
else: else:
logging.warning(f"Grad scale is small: {cur_grad_scale}") logging.warning(f"Grad scale is small: {cur_grad_scale}")
if params.multi_optim and batch_idx % params.accum_grads == 0: if params.multi_optim and batch_idx % params.accum_grads == 0:
set_batch_count(model, params.batch_idx_train) set_batch_count(model, params.batch_idx_train)
scheduler_enc.step_batch(params.batch_idx_train) scheduler_enc.step_batch(params.batch_idx_train)

7
egs/librispeech/ASR/run_v3.sh
View File

@ -36,13 +36,12 @@ else
--enable-spec-aug False \ --enable-spec-aug False \
--multi-optim True \ --multi-optim True \
--world-size 4 \ --world-size 4 \
--start-batch 34000 \
--num-epochs 30 \ --num-epochs 30 \
--start-epoch 6 \
--full-libri 1 \ --full-libri 1 \
--exp-dir ./pruned_transducer_stateless_d2v_v2/$1 \ --exp-dir ./pruned_transducer_stateless_d2v_v2/$1 \
--max-duration 150 \ --max-duration 150 \
--freeze-finetune-updates 2000 \ --freeze-finetune-updates 2000 # --use-fp16 1 \
--use-fp16 1 \
--peak-enc-lr 0.001 \ --peak-enc-lr 0.001 \
--peak-dec-lr 0.5 \ --peak-dec-lr 0.5 \
--accum-grads 3 \ --accum-grads 3 \
@ -55,3 +54,5 @@ else
--context-size 2 \ --context-size 2 \
--ctc-loss-scale 0.2 --ctc-loss-scale 0.2
fi fi
#--start-epoch 6 \