from local

This commit is contained in:
dohe0342 2022-12-12 15:48:09 +09:00
parent 704f197795
commit 3dc4283d58
2 changed files with 4 additions and 3 deletions

Binary file not shown.

View File

@ -1,6 +1,7 @@
git pull
export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
#export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
export CUDA_VISIBLE_DEVICES="0,1,2,3"
./pruned_transducer_stateless_d2v_v2/train.py \
--wandb true \
@ -21,8 +22,8 @@ export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
--encoder-type d2v \
--additional-block True \
--encoder-dim 768 \
--decoder-dim 768 \
--joiner-dim 768 \
--decoder-dim 512 \
--joiner-dim 512 \
--prune-range 20 \
--context-size 2 \
--ctc-loss-scale 0.2