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

Co-authored-by: Fangjun Kuang <csukuangfj@gmail.com>
This commit is contained in:
zr_jin 2024-04-08 12:03:24 +08:00 committed by GitHub
parent 8d05389bb2
commit 8347436e82
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1050,10 +1050,10 @@ def run(rank, world_size, args):
commonvoice = CommonVoiceAsrDataModule(args) commonvoice = CommonVoiceAsrDataModule(args)
if not args.use_validated_set: if args.use_validated_set:
train_cuts = commonvoice.train_cuts()
else:
train_cuts = commonvoice.validated_cuts() train_cuts = commonvoice.validated_cuts()
else:
train_cuts = commonvoice.train_cuts()
if args.use_invalidated_set: if args.use_invalidated_set:
train_cuts += commonvoice.invalidated_cuts() train_cuts += commonvoice.invalidated_cuts()