From b0719e068193fdb6f4669e83bd1ddbbac23d8e80 Mon Sep 17 00:00:00 2001 From: Wei-Chen-hub <1259566226@qq.com> Date: Thu, 23 May 2024 14:49:26 +0800 Subject: [PATCH] update --- tools/preprocess/synbody_preprocess_mp.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tools/preprocess/synbody_preprocess_mp.py b/tools/preprocess/synbody_preprocess_mp.py index a465690e..d73b8793 100644 --- a/tools/preprocess/synbody_preprocess_mp.py +++ b/tools/preprocess/synbody_preprocess_mp.py @@ -19,8 +19,12 @@ def process_npz_one(seq): - cmd = f'srun -p Zoetrope --gres=gpu:0 --cpus-per-task=4 -x SH-IDC1-10-198-8-[51,56,68,72,78,100,116,123] -N 1 ' \ - f'python tools/preprocess/synbody_preprocess.py --seq {seq} ' \ + # cmd = f'srun -p Zoetrope --gres=gpu:0 --cpus-per-task=4 -x SH-IDC1-10-198-8-[51,56,68,72,78,100,116,123] -N 1 ' \ + # f'python tools/preprocess/synbody_preprocess.py --seq {seq} ' \ + # f'--output_path {output_path} ' \ + # f'--prefix {prefix}' + + cmd = f'python tools/preprocess/synbody_preprocess.py --seq {seq} ' \ f'--output_path {output_path} ' \ f'--prefix {prefix}' @@ -88,7 +92,7 @@ def parse_args(): '--num_proc', required=False, type=int, - default=1, + default=4, help='num of processes') # parser.add_argument(