Update egs/commonvoice/ASR/pruned_transducer_stateless7/train.py

Co-authored-by: Fangjun Kuang <csukuangfj@gmail.com>
k2-fsa · Apr 8, 2024 · 8347436 · 8347436
1 parent 8d05389
commit 8347436
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/egs/commonvoice/ASR/pruned_transducer_stateless7/train.py b/egs/commonvoice/ASR/pruned_transducer_stateless7/train.py
@@ -1050,10 +1050,10 @@ def run(rank, world_size, args):
 
  commonvoice = CommonVoiceAsrDataModule(args)
 
- if not args.use_validated_set:
- train_cuts = commonvoice.train_cuts()
- else:
+ if args.use_validated_set:
  train_cuts = commonvoice.validated_cuts()
+ else:
+ train_cuts = commonvoice.train_cuts()
 
  if args.use_invalidated_set:
  train_cuts += commonvoice.invalidated_cuts()