From 604d0fdc211c53a0a3a9f66f0cf87f78fd740f30 Mon Sep 17 00:00:00 2001
From: Fabian Isensee <f.isensee@dkfz.de>
Date: Mon, 3 Jun 2024 10:15:31 +0200
Subject: [PATCH] use final checkpoint, we renamed the checkpoint in previous
 checkpoints

---
 nnunetv2/inference/JHU_inference.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nnunetv2/inference/JHU_inference.py b/nnunetv2/inference/JHU_inference.py
index d57c2a606..0933600a9 100644
--- 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