From 3b52e0cb9eee2830644eae4d89453e431cc509b2 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Thu, 29 May 2025 12:11:56 +0800 Subject: [PATCH] minor fixes --- egs/librispeech/ASR/zipformer/train_with_aug.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/egs/librispeech/ASR/zipformer/train_with_aug.py b/egs/librispeech/ASR/zipformer/train_with_aug.py index eb234bdcd..bec9f96c6 100755 --- a/egs/librispeech/ASR/zipformer/train_with_aug.py +++ b/egs/librispeech/ASR/zipformer/train_with_aug.py @@ -24,7 +24,7 @@ Usage: export CUDA_VISIBLE_DEVICES="0,1,2,3" # For non-streaming model training: -./zipformer/train.py \ +./zipformer/train_with_aug.py \ --world-size 4 \ --num-epochs 30 \ --start-epoch 1 \ @@ -34,7 +34,7 @@ export CUDA_VISIBLE_DEVICES="0,1,2,3" --max-duration 1000 # For streaming model training: -./zipformer/train.py \ +./zipformer/train_with_aug.py \ --world-size 4 \ --num-epochs 30 \ --start-epoch 1 \ @@ -70,7 +70,6 @@ from attention_decoder import AttentionDecoderModel from decoder import Decoder from joiner import Joiner from lhotse.cut import Cut -from lhotse.dataset import SpecAugment from lhotse.dataset.sampling.base import CutSampler from lhotse.utils import fix_random_seed from model import AsrModel