From 12a222aa4b70f145d91eb7acf4fce201ad35bdc7 Mon Sep 17 00:00:00 2001
From: Yifan Yang <64255737+yfyeung@users.noreply.github.com>
Date: Sun, 2 Apr 2023 16:32:43 +0800
Subject: [PATCH] Fix comments on the usage of train.py (#981)

---
 .../ASR/pruned_transducer_stateless2/train.py        | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/egs/gigaspeech/ASR/pruned_transducer_stateless2/train.py b/egs/gigaspeech/ASR/pruned_transducer_stateless2/train.py
index 9edc42b61..6a9f9f32f 100755
--- a/egs/gigaspeech/ASR/pruned_transducer_stateless2/train.py
+++ b/egs/gigaspeech/ASR/pruned_transducer_stateless2/train.py
@@ -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
 
 """