From 566ebaafe1b0d296d96867055abdd023454349d5 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Tue, 18 Apr 2023 13:56:18 +0900
Subject: [PATCH] from local

---
 egs/librispeech/ASR/full_ft.sh | 55 ++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)
 create mode 100755 egs/librispeech/ASR/full_ft.sh

diff --git a/egs/librispeech/ASR/full_ft.sh b/egs/librispeech/ASR/full_ft.sh
new file mode 100755
index 000000000..a373ff88c
--- /dev/null
+++ b/egs/librispeech/ASR/full_ft.sh
@@ -0,0 +1,55 @@
+git pull
+
+workstation="bear"
+
+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.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 
+else
+	export CUDA_VISIBLE_DEVICES="0,1,2,3"
+	./pruned_transducer_stateless_d2v_v2/train.py \
+		--wandb False \
+		--input-strategy AudioSamples \
+		--enable-spec-aug False \
+		--multi-optim True \
+		--world-size 4 \
+		--full-libri 1 \
+		--exp-dir ./pruned_transducer_stateless_d2v_v2/$1 \
+		--max-duration 150 \
+		--freeze-finetune-updates 0 \
+		--encoder-dim 768 \
+		--decoder-dim 768 \
+		--joiner-dim 768 \
+		--use-fp16 1 \
+		--peak-dec-lr 0.04175 \
+		--peak-enc-lr 0.0003859 \
+		--accum-grads 4 \
+		--encoder-type d2v \
+		--additional-block True \
+		--prune-range 10 \
+		--context-size 2 \
+		--ctc-loss-scale 0.2 
+fi