From 3eec5866de911499f79a9bd4ddef1d02bb0e4257 Mon Sep 17 00:00:00 2001
From: wasserth <j.wasserthal@gmx.de>
Date: Thu, 8 Feb 2024 17:08:18 +0100
Subject: [PATCH] add comment with adapted step size

---
 totalsegmentator/nnunet.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/totalsegmentator/nnunet.py b/totalsegmentator/nnunet.py
index aed5c9495..8e138efe0 100644
--- a/totalsegmentator/nnunet.py
+++ b/totalsegmentator/nnunet.py
@@ -180,6 +180,7 @@ def nnUNetv2_predict(dir_in, dir_out, task_id, model="3d_fullres", folds=None,
     else:
         device = torch.device('mps')
     step_size = 0.5
+    # step_size = 0.8  # overal speedup roughly 11%; for fast model no speedup; dice 0.001 worse
     disable_tta = not tta
     verbose = False
     save_probabilities = False