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-12 22:22:41 +09:00
parent 1be603f1d8
commit 271385f0f2
2 changed files with 27 additions and 26 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.

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

@ -1,32 +1,33 @@
git pull
#export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
export CUDA_VISIBLE_DEVICES="0,1,2,3"
workstation="whale"
./pruned_transducer_stateless_d2v_v2/train.py \
--wandb true \
--input-strategy AudioSamples \
--enable-spec-aug False \
--multi-optim True \
--world-size 4 \
--num-epochs 30 \
--start-epoch 1 \
--full-libri 1 \
--exp-dir ./pruned_transducer_stateless_d2v_v2/$1 \
--max-duration 150 \
--freeze-finetune-updates 2000 \
--use-fp16 1 \
--peak-enc-lr 0.001 \
--peak-dec-lr 0.05 \
--accum-grads 3 \
--encoder-type d2v \
--additional-block True \
--encoder-dim 768 \
--decoder-dim 768 \
--joiner-dim 768 \
--prune-range 10 \
--context-size 2 \
--ctc-loss-scale 0.2
if [ $workstation = "whale" ]; then
export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
./pruned_transducer_stateless_d2v_v2/train.py \
--wandb true \
--input-strategy AudioSamples \
--enable-spec-aug False \
--multi-optim True \
--world-size 8 \
--num-epochs 30 \
--start-epoch 1 \
--full-libri 1 \
--exp-dir ./pruned_transducer_stateless_d2v_v2/$1 \
--max-duration 250 \
--freeze-finetune-updates 2000 \
--use-fp16 1 \
--peak-enc-lr 0.001 \
--peak-dec-lr 0.05 \
--accum-grads 3 \
--encoder-type d2v \
--additional-block True \
--encoder-dim 768 \
--decoder-dim 768 \
--joiner-dim 768 \
--prune-range 10 \
--context-size 2 \
--ctc-loss-scale 0.2
#./pruned_transducer_stateless_d2v/train.py \
#--initial-lr 0.0003 \