created minimal memory use example

MHubAI · Nov 29, 2024 · 8ef8980 · 8ef8980
1 parent fc17471
commit 8ef8980
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/models/mrsegmentator/utils/MRSegmentatorMLRunner.py b/models/mrsegmentator/utils/MRSegmentatorMLRunner.py
@@ -34,7 +34,9 @@ def task(self, instance: Instance, in_data: InstanceData, out_data: InstanceData
         bash_command  = ["mrsegmentator"]
         bash_command += ["-i", in_data.abspath]
         bash_command += ["--outdir", tmp_dir]
-        bash_command += ["--nproc_export", "4"]
+        bash_command += ["--nproc_export", "1"]
+        bash_command += ["--nproc", "1"]
+        bash_command += ["--batchsize", "1"]
 
         if self.use_fast_mode:
             self.v("Running MRSegmentator in lower memory footprint mode ('--split_level', 1)")