Setting number of epochs #28

Activity

1. saikat-roy commented on Jul 30, 2024

   

   saikat-roy

   on Jul 30, 2024

   Contributor

   More actions

   Hi @toufiqmusah. You would have to create a new Trainer class by inheriting from nnUNetTrainerv2 and insert it into the constructor of the new Trainer class. Since I did this on nnUNet v1, there's no easier or more straightforward way to do this. This info should be available on the old nnUNet v1 branch of the nnUNet repository. Please let me know if you can't find it.
2. toufiqmusah commented on Aug 4, 2024

   

   toufiqmusah

   on Aug 4, 2024

   Author

   More actions

   I have been able to sort it out. Thank you very much

   Hi @toufiqmusah. You would have to create a new Trainer class by inheriting from nnUNetTrainerv2 and insert it into the constructor of the new Trainer class. Since I did this on nnUNet v1, there's no easier or more straightforward way to do this. This info should be available on the old nnUNet v1 branch of the nnUNet repository. Please let me know if you can't find it.