Fix break in fix_diagnostics mode
This commit is contained in:
parent
5a8cabd429
commit
e9668a5cfd
@ -794,7 +794,7 @@ def train_one_epoch(
|
|||||||
display_and_save_batch(batch, params=params)
|
display_and_save_batch(batch, params=params)
|
||||||
raise
|
raise
|
||||||
|
|
||||||
if params.print_diagnostics and batch_idx == 5:
|
if params.print_diagnostics and _batch_idx == 5:
|
||||||
return
|
return
|
||||||
|
|
||||||
if (
|
if (
|
||||||
|
|||||||
Reference in New Issue
Block a user