Project-MONAI / Project-MONAI/MONAI
Enable setting of training iteration in Trainers
@Warvito is already working on this.
Since Jun 23, 2022.
- Dominant language
- Python
- Stars
- 8.7k
- Forks
- 1.6k
- Avg merge
- 5d 1h
- Merged PRs (30d)
- 20
Description
Is your feature request related to a problem? Please describe.
Currently, SupervisedTrainer supports controlling the number of iterations by adjusting epoch_length and max_epochs. It would be nice to be able to set the number of iterations to be executed directly.
Describe the solution you'd like
Add a n_iterations argument (or similar) that allows overwriting epoch-based definitions of the training steps number of training steps to be executed.
Note, this should allow the training to resume from the final iteration if n_iterations is reached. Related to #4554.
Describe alternatives you've considered
Live with adjusting epoch_length and max_epochs but that seems confusing.
Additional context
Add any other context or screenshots about the feature request here.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.