minor fixes

This commit is contained in:
Fangjun Kuang 2025-05-29 12:11:56 +08:00
parent dc74705d20
commit 3b52e0cb9e

View File

@ -24,7 +24,7 @@ Usage:
export CUDA_VISIBLE_DEVICES="0,1,2,3" export CUDA_VISIBLE_DEVICES="0,1,2,3"
# For non-streaming model training: # For non-streaming model training:
./zipformer/train.py \ ./zipformer/train_with_aug.py \
--world-size 4 \ --world-size 4 \
--num-epochs 30 \ --num-epochs 30 \
--start-epoch 1 \ --start-epoch 1 \
@ -34,7 +34,7 @@ export CUDA_VISIBLE_DEVICES="0,1,2,3"
--max-duration 1000 --max-duration 1000
# For streaming model training: # For streaming model training:
./zipformer/train.py \ ./zipformer/train_with_aug.py \
--world-size 4 \ --world-size 4 \
--num-epochs 30 \ --num-epochs 30 \
--start-epoch 1 \ --start-epoch 1 \
@ -70,7 +70,6 @@ from attention_decoder import AttentionDecoderModel
from decoder import Decoder from decoder import Decoder
from joiner import Joiner from joiner import Joiner
from lhotse.cut import Cut from lhotse.cut import Cut
from lhotse.dataset import SpecAugment
from lhotse.dataset.sampling.base import CutSampler from lhotse.dataset.sampling.base import CutSampler
from lhotse.utils import fix_random_seed from lhotse.utils import fix_random_seed
from model import AsrModel from model import AsrModel