nnUNet
nnUNet copied to clipboard
Use real latest checkpoint in training
Solution for issue: https://github.com/MIC-DKFZ/nnUNet/issues/2563
Instead of loading the latest checkpoint when the flag --c is handled, we will filter to get the most recent checkpoint. This could be the latest, best, or final checkpoint.
@FabianIsensee please review it. As for me I always have to rename checkpoint if I stopped it before.
Hey I would prefer this logic would not use the file modification time as this can be flawed. If you want to make this logic more intelligent, please load the checkpoints and pick the one with the highest epoch