search

MIC-DKFZ / nnUNet

Apache License 2.0
5.95k stars 1.77k forks source link

Training with step instead of epoch #2606

Closed gogochen07 closed 2 days ago

gogochen07 commented 3 days ago

Hi, I am wondering if nnUNet v2 allow training with a certain number of optimization step instead of epoch? Thanks!

ykirchhoff commented 2 days ago

Hi @gogochen07,

per default nnUNet trains for 1000 epochs, where an epoch is defined as 250 iterations/optimization steps (see here). If I get your question correctly you should be able to achieve that by adjusting these numbers.

Best, Yannick

gogochen07 commented 2 days ago

Thanks so much! It works!