File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change @@ -645,10 +645,16 @@ def _test_check_triggered_events(
645645 n_iter_completed = max_epochs * epoch_length if max_epochs is not None else max_iters
646646
647647 if n_batch_started is None :
648- n_batch_started = max_epochs * epoch_length if max_epochs is not None else max_iters
648+ if data is None :
649+ n_batch_started = 0
650+ else :
651+ n_batch_started = max_epochs * epoch_length if max_epochs is not None else max_iters
649652
650653 if n_batch_completed is None :
651- n_batch_completed = max_epochs * epoch_length if max_epochs is not None else max_iters
654+ if data is None :
655+ n_batch_completed = 0
656+ else :
657+ n_batch_completed = max_epochs * epoch_length if max_epochs is not None else max_iters
652658
653659 if n_terminate is None :
654660 n_terminate = int (n_epoch_started != n_epoch_completed ) if max_iters is not None else 0
You can’t perform that action at this time.
0 commit comments