add comment with adapted step size

wasserth · Feb 8, 2024 · 3eec586 · 3eec586
1 parent f91dc99
commit 3eec586
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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