mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
9129a2b692
commit
ac1d5e63fd
Binary file not shown.
@ -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 \
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user