Fix: calculate num_iterations_per_epoch dynamically in nnUNetTrainer #2928
+23
−3
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Fixes: #2927
This update calculates the number of iterations per epoch dynamically instead of hardcoding it to 250.
The method
get_tr_and_val_datasetsin thennUNetTrainerclass was modified to update bothself.num_iterations_per_epochandself.num_val_iterations_per_epochbased on the dataset size and batch size.Additionally, progress bars were added to the training and validation loops to provide users with a clearer indication of progress during long training sessions.