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

Update finetune.py (#1304)

Browse Source
This commit is contained in:
zr_jin 2023-10-12 16:48:23 +08:00 committed by GitHub
parent 2b3c5d799f
commit 855492156a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/wenetspeech/ASR/pruned_transducer_stateless2/finetune.py
View File

@ -734,7 +734,7 @@ def train_one_epoch(
scaler.update() scaler.update()
optimizer.zero_grad() optimizer.zero_grad()
except: # noqa except: # noqa
display_and_save_batch(batch, params=params) display_and_save_batch(batch, params=params, graph_compiler=graph_compiler)
raise raise
if params.print_diagnostics and batch_idx == 5: if params.print_diagnostics and batch_idx == 5: