Fix print_diagnostics break statement

This commit is contained in:
Daniel Povey 2023-06-18 03:55:13 +08:00
parent e9668a5cfd
commit 70bd58c648

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 (