Increased memory for merge node.

CoBrALab · Jan 21, 2020 · 6d4ed28 · 6d4ed28
1 parent e371af9
commit 6d4ed28
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/rabies/preprocess_bold_pkg/resampling.py b/rabies/preprocess_bold_pkg/resampling.py
@@ -21,14 +21,14 @@ def init_bold_preproc_trans_wf(resampling_dim, name='bold_preproc_trans_wf'):
         name='outputnode')
 
 
-    bold_transform = pe.Node(slice_applyTransforms(), name='bold_transform', mem_gb=3)
+    bold_transform = pe.Node(slice_applyTransforms(), name='bold_transform')
     bold_transform.inputs.apply_motcorr = True
     bold_transform.inputs.resampling_dim = resampling_dim
     bold_transform.inputs.data_type = os.environ["rabies_data_type"]
 
-    merge = pe.Node(Merge(), name='merge', mem_gb=3)
+    merge = pe.Node(Merge(), name='merge', mem_gb=5)
     merge.inputs.data_type = os.environ["rabies_data_type"]
-    merge.plugin_args = {'qsub_args': '-pe smp %s' % (str(3*int(os.environ["min_proc"]))), 'overwrite': True}
+    merge.plugin_args = {'qsub_args': '-pe smp %s' % (str(4*int(os.environ["min_proc"]))), 'overwrite': True}
 
     # Generate a new BOLD reference
     bold_reference_wf = init_bold_reference_wf()
@@ -66,15 +66,15 @@ def init_bold_commonspace_trans_wf(resampling_dim, name='bold_commonspace_trans_
         niu.IdentityInterface(fields=['bold', 'bold_ref', 'brain_mask', 'WM_mask', 'CSF_mask', 'labels']),
         name='outputnode')
 
-    bold_transform = pe.Node(slice_applyTransforms(), name='bold_transform', mem_gb=3)
+    bold_transform = pe.Node(slice_applyTransforms(), name='bold_transform')
     bold_transform.inputs.apply_motcorr = True
     bold_transform.inputs.ref_file = os.environ["template_anat"]
     bold_transform.inputs.resampling_dim = resampling_dim
     bold_transform.inputs.data_type = os.environ["rabies_data_type"]
 
-    merge = pe.Node(Merge(), name='merge', mem_gb=3)
+    merge = pe.Node(Merge(), name='merge', mem_gb=5)
     merge.inputs.data_type = os.environ["rabies_data_type"]
-    merge.plugin_args = {'qsub_args': '-pe smp %s' % (str(3*int(os.environ["min_proc"]))), 'overwrite': True}
+    merge.plugin_args = {'qsub_args': '-pe smp %s' % (str(4*int(os.environ["min_proc"]))), 'overwrite': True}
 
     # Generate a new BOLD reference
     bold_reference_wf = init_bold_reference_wf()