fix missing default value for nnunet model configurable in ProstateRu…

…nner module
MHubAI · Jan 12, 2024 · 1115f31 · 1115f31
1 parent 46f6357
commit 1115f31
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/models/nnunet_prostate_zonal_task05/utils/ProstateRunner.py b/models/nnunet_prostate_zonal_task05/utils/ProstateRunner.py
@@ -2,7 +2,7 @@
 from mhubio.core import Module, Instance, InstanceData, IO
 
 @IO.Config('use_tta', bool, False, the='flag to enable test time augmentation')
-@IO.Config('nnunet_model', str, None, the='nnunet model name (2d, 3d_lowres, 3d_fullres, 3d_cascade_fullres)')
+@IO.Config('nnunet_model', str, '3d_fullres', the='nnunet model name (2d, 3d_lowres, 3d_fullres, 3d_cascade_fullres)')
 class ProstateRunner(Module):
 
     use_tta: bool
@@ -45,5 +45,7 @@ def task(self, instance: Instance, T2: InstanceData, ADC: InstanceData, P: Insta
         if not self.use_tta:
             bash_command += ["--disable_tta"]
 
+        print(bash_command)
+
         # run command
         self.subprocess(bash_command, text=True)