mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-01-04 14:33:44 +09:00
parent 2a0ae840ec
commit f06af120fc
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/pruned_transducer_stateless5/.train.py.swp
View File

Binary file not shown.

10
egs/librispeech/ASR/pruned_transducer_stateless5/train.py
View File

@ -1024,9 +1024,9 @@ def run(rank, world_size, args):
# an utterance duration distribution for your dataset to select
# the threshold
if c.duration < 1.0 or c.duration > 20.0:
logging.warning(
f"Exclude cut with ID {c.id} from training. Duration: {c.duration}"
)
#logging.warning(
# f"Exclude cut with ID {c.id} from training. Duration: {c.duration}"
#)
return False
# In pruned RNN-T, we require that T >= S
@ -1067,7 +1067,8 @@ def run(rank, world_size, args):
valid_cuts = librispeech.dev_clean_cuts()
valid_cuts += librispeech.dev_other_cuts()
valid_dl = librispeech.valid_dataloaders(valid_cuts)
'''
if params.start_batch <= 0 and not params.print_diagnostics:
scan_pessimistic_batches_for_oom(
model=model,
@ -1077,6 +1078,7 @@ def run(rank, world_size, args):
params=params,
warmup=0.0 if params.start_epoch == 1 else 1.0,
)
'''
scaler = GradScaler(enabled=params.use_fp16)
if checkpoints and "grad_scaler" in checkpoints: