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 00:24:44 +09:00
parent 6f91da4ed4
commit 0ad344126c
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

Binary file not shown.

4
egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py
View File

@ -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: