From 8347436e82656eefa031f33863045af6b4d5a416 Mon Sep 17 00:00:00 2001 From: zr_jin Date: Mon, 8 Apr 2024 12:03:24 +0800 Subject: [PATCH] Update egs/commonvoice/ASR/pruned_transducer_stateless7/train.py Co-authored-by: Fangjun Kuang --- egs/commonvoice/ASR/pruned_transducer_stateless7/train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/egs/commonvoice/ASR/pruned_transducer_stateless7/train.py b/egs/commonvoice/ASR/pruned_transducer_stateless7/train.py index cae6e6429b..5e98084ec0 100755 --- 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()