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

style check

Browse Source
This commit is contained in:
luomingshuang 2022-06-26 20:41:21 +08:00
parent 77b4224686
commit e038fed516
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -830,7 +830,9 @@ def train_one_epoch(
scaler.update()
optimizer.zero_grad()
except: # noqa
display_and_save_batch(batch, params=params, graph_compiler=graph_compiler)
display_and_save_batch(
batch, params=params, graph_compiler=graph_compiler
)
raise
if params.print_diagnostics and batch_idx == 5:
@ -929,7 +931,7 @@ def run(rank, world_size, args):
"""
params = get_params()
params.update(vars(args))
fix_random_seed(params.seed)
if world_size > 1:
setup_dist(rank, world_size, params.master_port)
@ -1169,7 +1171,9 @@ def scan_pessimistic_batches_for_oom(
f"Failing criterion: {criterion} "
f"(={crit_values[criterion]}) ..."
)
display_and_save_batch(batch, params=params, graph_compiler=graph_compiler)
display_and_save_batch(
batch, params=params, graph_compiler=graph_compiler
)
raise