use final checkpoint, we renamed the checkpoint in previous checkpoints

MIC-DKFZ · Jun 3, 2024 · 604d0fd · 604d0fd
1 parent 80f7c3d
commit 604d0fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nnunetv2/inference/JHU_inference.py b/nnunetv2/inference/JHU_inference.py
@@ -176,7 +176,7 @@ def predict_from_data_iterator(self,
     predictor.initialize_from_trained_model_folder(
         args.model,
         ('all', ),
-        'checkpoint_latest.pth'
+        'checkpoint_final.pth'
     )
 
     # we need to create list of list of input files