mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 10:02:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix comments on the usage of train.py (#981)

Browse Source
This commit is contained in:
Yifan Yang 2023-04-02 16:32:43 +08:00 committed by GitHub
parent a632b24c35
commit 12a222aa4b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
egs/gigaspeech/ASR/pruned_transducer_stateless2/train.py
View File

@ -19,26 +19,24 @@
"""
Usage:
export CUDA_VISIBLE_DEVICES="0,1,2,3"
export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
./pruned_transducer_stateless2/train.py \
--world-size 4 \
--world-size 8 \
--num-epochs 30 \
--start-epoch 0 \
--exp-dir pruned_transducer_stateless2/exp \
--full-libri 1 \
--max-duration 300
--max-duration 120
# For mix precision training:
./pruned_transducer_stateless2/train.py \
--world-size 4 \
--world-size 8 \
--num-epochs 30 \
--start-epoch 0 \
--use_fp16 1 \
--exp-dir pruned_transducer_stateless2/exp \
--full-libri 1 \
--max-duration 550
--max-duration 200
"""