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

Fixes after review.

Browse Source
This commit is contained in:
Fangjun Kuang 2022-05-05 12:47:24 +08:00
parent ce885c6a67
commit a0dbfba77d
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
egs/librispeech/ASR/pruned_transducer_stateless2/train.py
View File

@ -691,9 +691,8 @@ def train_one_epoch(
scaler.step(optimizer) scaler.step(optimizer)
scaler.update() scaler.update()
optimizer.zero_grad() optimizer.zero_grad()
except RuntimeError as e: except:
if "CUDA out of memory" in str(e): display_and_save_batch(batch, params=params, sp=sp)
display_and_save_batch(batch, params=params, sp=sp)
raise raise
if params.print_diagnostics and batch_idx == 5: if params.print_diagnostics and batch_idx == 5: