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
1be603f1d8
commit
271385f0f2
Binary file not shown.
@ -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"
|
|
||||||
|
|
||||||
|
if [ $workstation = "whale" ]; then
|
||||||
|
export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
|
||||||
./pruned_transducer_stateless_d2v_v2/train.py \
|
./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 \
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user