Fix break in fix_diagnostics mode

This commit is contained in:
Daniel Povey 2023-06-18 03:36:00 +08:00
parent 5a8cabd429
commit e9668a5cfd

View File

@ -794,7 +794,7 @@ def train_one_epoch(
display_and_save_batch(batch, params=params)
raise
if params.print_diagnostics and batch_idx == 5:
if params.print_diagnostics and _batch_idx == 5:
return
if (