-
-
Notifications
You must be signed in to change notification settings - Fork 654
Closed
Description
🚀 Feature
The following code may be fail if max_epochs is None:
ignite/ignite/engine/engine.py
Lines 839 to 842 in a756f9c
| "Current iteration: {} vs Total iterations to run : {}".format( | |
| self.state.iteration, | |
| self.state.epoch_length * self.state.max_epochs, # type: ignore[operator] | |
| ) |
ignite/ignite/engine/engine.py
Lines 562 to 566 in a756f9c
| is_done_count = ( | |
| state.epoch_length is not None | |
| and state.iteration >= state.epoch_length * state.max_epochs # type: ignore[operator] | |
| ) | |
| is_done_epochs = state.max_epochs is not None and state.epoch >= state.max_epochs |
Maybe, we have to add additional check of max_epochs is not None and remove # type: ignore[operator].
@gruebel what do you think ?
Metadata
Metadata
Assignees
Labels
No labels