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 2022-12-09 17:28:09 +09:00
parent bd2eea831e
commit 5a56f4adfb
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_d2v_v2/.train.py.swp
View File

Binary file not shown.

2
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
View File

@ -1124,6 +1124,7 @@ def run(rank, world_size, args):
valid_cuts += librispeech.dev_other_cuts()
valid_dl = librispeech.valid_dataloaders(valid_cuts)
'''
if not params.print_diagnostics:
scan_pessimistic_batches_for_oom(
model=model,
@ -1132,6 +1133,7 @@ def run(rank, world_size, args):
sp=sp,
params=params,
)
'''
scaler = GradScaler(enabled=params.use_fp16, init_scale=1.0)
if checkpoints and "grad_scaler" in checkpoints: