mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-27 02:34:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

add remove long short

Browse Source
This commit is contained in:
Yuekai Zhang 2024-01-26 10:23:26 +08:00
parent bb07b65e45
commit c19891ee8e
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
egs/wenetspeech/ASR/whisper/train.py
View File

@ -834,8 +834,9 @@ def run(rank, world_size, args):
# ) # )
return False return False
return True return True
train_cuts = wenetspeech.train_cuts()
train_dl = wenetspeech.train_dataloaders(wenetspeech.train_cuts(remove_short_and_long_utt)) train_cuts = train_cuts.filter(remove_short_and_long_utt)
train_dl = wenetspeech.train_dataloaders(train_cuts)
valid_dl = wenetspeech.valid_dataloaders(wenetspeech.valid_cuts()) valid_dl = wenetspeech.valid_dataloaders(wenetspeech.valid_cuts())
scaler = GradScaler(enabled=params.use_fp16, init_scale=1.0) scaler = GradScaler(enabled=params.use_fp16, init_scale=1.0)