from local

This commit is contained in:
dohe0342 2022-12-27 10:27:45 +09:00
parent 9129a2b692
commit ac1d5e63fd
2 changed files with 1 additions and 6 deletions

View File

@ -16,17 +16,12 @@ if [ $workstation = "whale" ]; then
--exp-dir ./pruned_transducer_stateless_d2v_v2/$1 \ --exp-dir ./pruned_transducer_stateless_d2v_v2/$1 \
--max-duration 250 \ --max-duration 250 \
--use-fp16 1 \ --use-fp16 1 \
--peak-enc-lr 0.03 \
--peak-dec-lr 0.1 \
--accum-grads 1 \
--encoder-type d2v \ --encoder-type d2v \
--additional-block True \ --additional-block True \
--encoder-dim 768 \ --encoder-dim 768 \
--decoder-dim 768 \ --decoder-dim 768 \
--joiner-dim 768 \ --joiner-dim 768 \
--prune-range 10 \ --prune-range 10
--context-size 2 \
--ctc-loss-scale 0.2
else else
export CUDA_VISIBLE_DEVICES="0,1,2,3" export CUDA_VISIBLE_DEVICES="0,1,2,3"
./pruned_transducer_stateless_d2v_v2/train.py \ ./pruned_transducer_stateless_d2v_v2/train.py \