from local
This commit is contained in:
parent
704f197795
commit
3dc4283d58
Binary file not shown.
@ -1,6 +1,7 @@
|
|||||||
git pull
|
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 \
|
./pruned_transducer_stateless_d2v_v2/train.py \
|
||||||
--wandb true \
|
--wandb true \
|
||||||
@ -21,8 +22,8 @@ export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
|
|||||||
--encoder-type d2v \
|
--encoder-type d2v \
|
||||||
--additional-block True \
|
--additional-block True \
|
||||||
--encoder-dim 768 \
|
--encoder-dim 768 \
|
||||||
--decoder-dim 768 \
|
--decoder-dim 512 \
|
||||||
--joiner-dim 768 \
|
--joiner-dim 512 \
|
||||||
--prune-range 20 \
|
--prune-range 20 \
|
||||||
--context-size 2 \
|
--context-size 2 \
|
||||||
--ctc-loss-scale 0.2
|
--ctc-loss-scale 0.2
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user