from local

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

Binary file not shown.

View File

@ -1,19 +1,20 @@
git pull git pull
#export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7" workstation="whale"
export CUDA_VISIBLE_DEVICES="0,1,2,3"
./pruned_transducer_stateless_d2v_v2/train.py \ 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 \ --wandb true \
--input-strategy AudioSamples \ --input-strategy AudioSamples \
--enable-spec-aug False \ --enable-spec-aug False \
--multi-optim True \ --multi-optim True \
--world-size 4 \ --world-size 8 \
--num-epochs 30 \ --num-epochs 30 \
--start-epoch 1 \ --start-epoch 1 \
--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 250 \
--freeze-finetune-updates 2000 \ --freeze-finetune-updates 2000 \
--use-fp16 1 \ --use-fp16 1 \
--peak-enc-lr 0.001 \ --peak-enc-lr 0.001 \