⚡ Increase memory estimates by 0.5GB on overrunning nodes

As seen in #1457 (comment)
FCP-INDI · Mar 10, 2021 · bdd1392 · bdd1392
1 parent 3999173
commit bdd1392
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/CPAC/registration/registration.py b/CPAC/registration/registration.py
@@ -2272,7 +2272,8 @@ def warp_timeseries_to_T1template(wf, cfg, strat_pool, pipe_num, opt=None):
 
     apply_xfm = apply_transform(f'warp_ts_to_T1template_{pipe_num}', reg_tool,
                                 time_series=True, num_cpus=num_cpus,
-                                num_ants_cores=num_ants_cores)
+                                num_ants_cores=num_ants_cores,
+                                mem_gb=2.5)
 
     if reg_tool == 'ants':
         apply_xfm.inputs.inputspec.interpolation = cfg.registration_workflows[
@@ -2487,9 +2488,10 @@ def warp_timeseries_to_EPItemplate(wf, cfg, strat_pool, pipe_num, opt=None):
 
     num_ants_cores = cfg.pipeline_setup['system_config']['num_ants_threads']
 
-    apply_xfm = apply_transform('warp_ts_to_EPItemplate', reg_tool,
+    apply_xfm = apply_transform(f'warp_ts_to_EPItemplate_{pipe_num}', reg_tool,
                                 time_series=True, num_cpus=num_cpus,
-                                num_ants_cores=num_ants_cores)
+                                num_ants_cores=num_ants_cores,
+                                mem_gb=2.5)
 
     if reg_tool == 'ants':
         apply_xfm.inputs.inputspec.interpolation = cfg.registration_workflows[

diff --git a/CPAC/seg_preproc/seg_preproc.py b/CPAC/seg_preproc/seg_preproc.py
@@ -575,8 +575,7 @@ def tissue_seg_fsl_fast(wf, cfg, strat_pool, pipe_num, opt=None):
     #  'probability_maps' output is a list of individual probability maps
     #      triggered by 'probability_maps' boolean input (-p)
 
-    segment = pe.Node(interface=fsl.FAST(), name=f'segment_{pipe_num}',
-                      mem_gb=1.5)
+    segment = pe.Node(interface=fsl.FAST(), name=f'segment_{pipe_num}')
     segment.inputs.img_type = 1
     segment.inputs.segments = True
     segment.inputs.probability_maps = True

diff --git a/CPAC/vmhc/vmhc.py b/CPAC/vmhc/vmhc.py
@@ -81,7 +81,8 @@ def warp_timeseries_to_sym_template(wf, cfg, strat_pool, pipe_num, opt=None):
 
     apply_xfm = apply_transform(f'warp_ts_to_sym_template_{pipe_num}',
                                 reg_tool, time_series=True, num_cpus=num_cpus,
-                                num_ants_cores=num_ants_cores)
+                                num_ants_cores=num_ants_cores,
+                                mem_gb=2.5)
 
     if reg_tool == 'ants':
         apply_xfm.inputs.inputspec.interpolation = cfg.registration_workflows[