From c0e311a0d6a6b857019af4cecdb4364c4f9f143b Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Mon, 12 Dec 2022 22:24:14 +0900 Subject: [PATCH] from local --- egs/librispeech/ASR/.run_v3.sh.swp | Bin 12288 -> 12288 bytes egs/librispeech/ASR/run_v3.sh | 32 ++++++++++++++++++++++++----- 2 files changed, 27 insertions(+), 5 deletions(-) diff --git a/egs/librispeech/ASR/.run_v3.sh.swp b/egs/librispeech/ASR/.run_v3.sh.swp index 2cfd10ad88fc9cb44c2f1431501cfee08db7a210..24df770a1b40d9a523e7ea4aa0327e93de6d1367 100644 GIT binary patch delta 242 zcmXBNtqQ_m7zNQ!1@L>o|us{JBUU+!G74{Ip?{xz%I8X&7PKsqI sIB0Tk*I1DhUAK96Y7ZtoeasjAsm*#b9-K5LID^>Gmx9PgbB=Zsz9KF%bN~PV delta 142 zcmZojXh;xEG6?hZRj|}EU;qLE28Lkm>B+SlMSt@%DsL7P_{`6%$iu*J5{N;P%98~Z z>g5c085o#=_%%@U6jb&Yklq2r+j%xC3UKpGw$$Qf2I`qytEDhGLW_^5NWnnQz`(#* cAqHqjy`CbEeo0YcW}aR_B~WCuf$jw+0F4bF&j0`b diff --git a/egs/librispeech/ASR/run_v3.sh b/egs/librispeech/ASR/run_v3.sh index b80b38a96..8989a8427 100755 --- a/egs/librispeech/ASR/run_v3.sh +++ b/egs/librispeech/ASR/run_v3.sh @@ -28,8 +28,30 @@ if [ $workstation = "whale" ]; then --prune-range 10 \ --context-size 2 \ --ctc-loss-scale 0.2 - -#./pruned_transducer_stateless_d2v/train.py \ -#--initial-lr 0.0003 \ -#--decoding-method greedy_search - +else + 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 200 \ + --freeze-finetune-updates 2000 \ + --use-fp16 1 \ + --peak-enc-lr 0.03 \ + --peak-dec-lr 0.1 \ + --accum-grads 1 \ + --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 +fi