add model name to configs

main/config/config_ft_3dpw.py

@@ -18,7 +18,7 @@ bbox_ratio = 1.2
 # continue
 continue_train = True
 start_over = True
-pretrained_model_path = '../output/train_exp118_20230607_192422/model_dump/snapshot_9.pth.tar'
+pretrained_model_path = '../path_to_smpler_x_h32/snapshot.pth.tar'
 
 # dataset setting
 agora_fix_betas = True

main/config/config_ft_agora.py

@@ -18,7 +18,7 @@ bbox_ratio = 1.2
 # continue
 continue_train = True
 start_over = True
-pretrained_model_path = '../output/train_exp118_20230607_192422/model_dump/snapshot_9.pth.tar'
+pretrained_model_path = '../path_to_smpler_x_h32/snapshot.pth.tar'
 
 # dataset setting
 agora_fix_betas = True
@@ -30,7 +30,7 @@ dataset_list = ['Human36M', 'MSCOCO', 'MPII', 'AGORA', 'EHF', 'SynBody', 'GTA_Hu
     'EgoBody_Egocentric', 'EgoBody_Kinect', 'UBody', 'PW3D', 'MuCo', 'PROX']
 trainset_3d = ['AGORA'] 
 trainset_2d = []
-trainset_humandata = ['GTA_Human2', 'BEDLAM', 'SynBody'] 
+trainset_humandata = ['GTA_Human2', 'BEDLAM', 'SynBody', 'SPEC'] 
 testset = 'EHF'
 
 use_cache = True

main/config/config_ft_arctic.py

@@ -18,7 +18,7 @@ bbox_ratio = 1.2
 # continue
 continue_train = True
 start_over = True
-pretrained_model_path = '../output/train_exp118_20230607_192422/model_dump/snapshot_9.pth.tar'
+pretrained_model_path = '../path_to_smpler_x_h32/snapshot.pth.tar'
 # dataset setting
 agora_fix_betas = True
 agora_fix_global_orient_transl = True

main/config/config_ft_egobody.py

@@ -18,7 +18,7 @@ bbox_ratio = 1.2
 # continue
 continue_train = True
 start_over = True
-pretrained_model_path = '../output/train_exp118_20230607_192422/model_dump/snapshot_9.pth.tar'
+pretrained_model_path = '../path_to_smpler_x_h32/snapshot.pth.tar'
 # dataset setting
 agora_fix_betas = True
 agora_fix_global_orient_transl = True

main/config/config_ft_renbody.py

@@ -18,7 +18,7 @@ bbox_ratio = 1.2
 # continue
 continue_train = True
 start_over = True
-pretrained_model_path = '../output/train_exp118_20230607_192422/model_dump/snapshot_9.pth.tar'
+pretrained_model_path = '../path_to_smpler_x_h32/snapshot.pth.tar'
 # dataset setting
 agora_fix_betas = True
 agora_fix_global_orient_transl = True
@@ -29,7 +29,7 @@ dataset_list = ['Human36M', 'MSCOCO', 'MPII', 'AGORA', 'EHF', 'SynBody', 'GTA_Hu
     'EgoBody_Egocentric', 'EgoBody_Kinect', 'UBody', 'PW3D', 'MuCo', 'PROX']
 trainset_3d = [] 
 trainset_2d = []
-trainset_humandata = ['RenBody_HiRes'] 
+trainset_humandata = ['RenBody_HiRes', 'RenBody', 'BEDLAM', 'SynBody', 'CHI3D'] 
 testset = 'EHF'
 
 use_cache = True

main/config/config_ft_ubody.py

@@ -18,7 +18,7 @@ bbox_ratio = 1.2
 # continue
 continue_train = True
 start_over = True
-pretrained_model_path = '../output/train_exp118_20230607_192422/model_dump/snapshot_9.pth.tar'
+pretrained_model_path = '../path_to_smpler_x_h32/snapshot.pth.tar'
 
 # dataset setting
 agora_fix_betas = True

main/config/config_smpler_x_b20.py

@@ -17,7 +17,6 @@ bbox_ratio = 1.2
 # continue
 continue_train = False
 start_over = True
-# pretrained_model_path = '../output/train_exp78_20230526_105124/model_dump/snapshot_4.pth.tar'
 
 # dataset setting
 agora_fix_betas = True

main/config/config_smpler_x_b32.py

@@ -15,9 +15,8 @@ syncbn = True
 bbox_ratio = 1.2
 
 # continue
-continue_train = True
-start_over = False
-pretrained_model_path = '../output/train_exp110_20230602_211349/model_dump/snapshot_0.pth.tar'
+continue_train = False
+start_over = True
 
 # dataset setting
 agora_fix_betas = True

main/config/config_smpler_x_b5.py

@@ -17,7 +17,6 @@ bbox_ratio = 1.2
 # continue
 continue_train = False
 start_over = True
-# pretrained_model_path = '../output/train_exp78_20230526_105124/model_dump/snapshot_4.pth.tar'
 
 # dataset setting
 agora_fix_betas = True

main/config/config_smpler_x_h10.py

@@ -17,7 +17,6 @@ bbox_ratio = 1.2
 # continue
 continue_train = False
 start_over = True
-# pretrained_model_path = '../output/train_exp78_20230526_105124/model_dump/snapshot_4.pth.tar'
 
 # dataset setting
 agora_fix_betas = True

main/config/config_smpler_x_h20.py

@@ -17,7 +17,6 @@ bbox_ratio = 1.2
 # continue
 continue_train = False
 start_over = True
-# pretrained_model_path = '../output/train_exp78_20230526_105124/model_dump/snapshot_4.pth.tar'
 
 # dataset setting
 agora_fix_betas = True

main/config/config_smpler_x_h32.py

@@ -17,7 +17,6 @@ bbox_ratio = 1.2
 # continue
 continue_train = False
 start_over = True
-# pretrained_model_path = '../output/train_exp118_20230602_211120/model_dump/snapshot_0.pth.tar'
 
 # dataset setting
 agora_fix_betas = True

main/config/config_smpler_x_h5.py

@@ -17,7 +17,6 @@ bbox_ratio = 1.2
 # continue
 continue_train = False
 start_over = True
-# pretrained_model_path = '../output/train_exp78_20230526_105124/model_dump/snapshot_4.pth.tar'
 
 # dataset setting
 agora_fix_betas = True

main/config/config_smpler_x_l10.py

@@ -17,7 +17,6 @@ bbox_ratio = 1.2
 # continue
 continue_train = False
 start_over = True
-# pretrained_model_path = '../output/train_exp78_20230526_105124/model_dump/snapshot_4.pth.tar'
 
 # dataset setting
 agora_fix_betas = True

main/config/config_smpler_x_l20.py

@@ -17,7 +17,6 @@ bbox_ratio = 1.2
 # continue
 continue_train = False
 start_over = True
-# pretrained_model_path = '../output/train_exp78_20230526_105124/model_dump/snapshot_4.pth.tar'
 
 # dataset setting
 agora_fix_betas = True

main/config/config_smpler_x_l32.py

@@ -15,9 +15,8 @@ syncbn = True
 bbox_ratio = 1.2
 
 # continue
-continue_train = True
-start_over = False
-pretrained_model_path = '../output/train_exp114_20230602_194715/model_dump/snapshot_0.pth.tar'
+continue_train = False
+start_over = True
 
 # dataset setting
 agora_fix_betas = True

main/config/config_smpler_x_l5.py

@@ -17,7 +17,6 @@ bbox_ratio = 1.2
 # continue
 continue_train = False
 start_over = True
-# pretrained_model_path = '../output/train_exp111_20230602_184623/model_dump/snapshot_4.pth.tar'
 
 # dataset setting
 agora_fix_betas = True

main/config/config_smpler_x_s10.py

@@ -17,7 +17,6 @@ bbox_ratio = 1.2
 # continue
 continue_train = False
 start_over = True
-# pretrained_model_path = '../output/train_exp78_20230526_105124/model_dump/snapshot_4.pth.tar'
 
 # dataset setting
 agora_fix_betas = True

main/config/config_smpler_x_s20.py

@@ -15,9 +15,8 @@ syncbn = True
 bbox_ratio = 1.2
 
 # continue
-continue_train = True
-start_over = False
-pretrained_model_path = '../output/train_exp97_20230527_164023/model_dump/snapshot_0.pth.tar'
+continue_train = False
+start_over = True
 
 # dataset setting
 agora_fix_betas = True

main/config/config_smpler_x_s32.py

@@ -15,9 +15,8 @@ syncbn = True
 bbox_ratio = 1.2
 
 # continue
-continue_train = True
-start_over = False
-pretrained_model_path = '../output/train_exp106_20230602_211720/model_dump/snapshot_0.pth.tar'
+continue_train = False
+start_over = True
 
 # dataset setting
 agora_fix_betas = True

main/slurm_test.sh

@@ -1,4 +1,5 @@
 #!/usr/bin/env bash
+#!/usr/bin/env bash
 
 set -x
 
@@ -8,84 +9,103 @@ GPUS=$2
 RES_PATH=$3
 CKPT=$4
 
-# model_path=../output/train_gta_synbody_ft_20230410_132110/model_dump/snapshot_2.pth.tar
-
 GPUS_PER_NODE=$((${GPUS}<8?${GPUS}:8))
 CPUS_PER_TASK=${CPUS_PER_TASK:-2}
 SRUN_ARGS=${SRUN_ARGS:-""}
 
-# PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
-# srun -p ${PARTITION} \
-#     --job-name=${JOB_NAME} \
-#     --gres=gpu:${GPUS_PER_NODE} \
-#     --ntasks-per-node=1 \
-#     --cpus-per-task=${CPUS_PER_TASK} \
-#     --kill-on-bad-exit=1 \
-#     ${SRUN_ARGS} \
-#     python test.py \
-#         --num_gpus ${GPUS_PER_NODE} \
-#         --exp_name output/test_${JOB_NAME}_ep${CKPT}_PW3D \
-#         --result_path ${RES_PATH} \
-#         --ckpt_idx ${CKPT} \
-#         --testset PW3D \
-#         # --agora_benchmark agora_model_val
-
-# PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
-# srun -p ${PARTITION} \
-#     --job-name=${JOB_NAME} \
-#     --gres=gpu:${GPUS_PER_NODE} \
-#     --ntasks-per-node=1 \
-#     --cpus-per-task=${CPUS_PER_TASK} \
-#     --kill-on-bad-exit=1 \
-#     ${SRUN_ARGS} \
-#     python test.py \
-#         --num_gpus ${GPUS_PER_NODE} \
-#         --exp_name output/test_${JOB_NAME}_ep${CKPT}_EgoBody_Egocentric \
-#         --result_path ${RES_PATH} \
-#         --ckpt_idx ${CKPT} \
-#         --testset EgoBody_Egocentric \
-#         # --agora_benchmark agora_model_val
-
-# PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
-# srun -p ${PARTITION} \
-#     --job-name=${JOB_NAME} \
-#     --gres=gpu:${GPUS_PER_NODE} \
-#     --ntasks-per-node=1 \
-#     --cpus-per-task=${CPUS_PER_TASK} \
-#     --kill-on-bad-exit=1 \
-#     ${SRUN_ARGS} \
-#     python test.py \
-#         --num_gpus ${GPUS_PER_NODE} \
-#         --exp_name output/test_${JOB_NAME}_ep${CKPT}_UBody \
-#         --result_path ${RES_PATH} \
-#         --ckpt_idx ${CKPT} \
-#         --testset UBody \
-#         # --agora_benchmark agora_model_val
-
-# PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
-# srun -p ${PARTITION} \
-#     --job-name=${JOB_NAME} \
-#     --gres=gpu:${GPUS_PER_NODE} \
-#     --ntasks-per-node=1 \
-#     --cpus-per-task=${CPUS_PER_TASK} \
-#     --kill-on-bad-exit=1 \
-#     ${SRUN_ARGS} \
-#     python test.py \
-#         --num_gpus ${GPUS_PER_NODE} \
-#         --exp_name output/test_${JOB_NAME}_ep${CKPT}_SHAPY \
-#         --result_path ${RES_PATH} \
-#         --ckpt_idx ${CKPT} \
-#         --testset SHAPY \
-#         --shapy_eval_split test
-
 PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
 srun -p ${PARTITION} \
     --job-name=${JOB_NAME} \
     --gres=gpu:${GPUS_PER_NODE} \
     --ntasks-per-node=1 \
     --cpus-per-task=${CPUS_PER_TASK} \
+    --quotatype=auto \
+    --kill-on-bad-exit=1 \
+    ${SRUN_ARGS} \
+    python test.py \
+        --num_gpus ${GPUS_PER_NODE} \
+        --exp_name output/test_${JOB_NAME}_ep${CKPT}_PW3D \
+        --result_path ${RES_PATH} \
+        --ckpt_idx ${CKPT} \
+        --testset PW3D \
+        --agora_benchmark agora_model_val
+
+PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
+srun -p ${PARTITION} \
+    --job-name=${JOB_NAME} \
+    --gres=gpu:${GPUS_PER_NODE} \
+    --ntasks-per-node=1 \
+    --cpus-per-task=${CPUS_PER_TASK} \
+    --quotatype=auto \
+    --kill-on-bad-exit=1 \
+    ${SRUN_ARGS} \
+    python test.py \
+        --num_gpus ${GPUS_PER_NODE} \
+        --exp_name output/test_${JOB_NAME}_ep${CKPT}_EgoBody_Egocentric \
+        --result_path ${RES_PATH} \
+        --ckpt_idx ${CKPT} \
+        --testset EgoBody_Egocentric \
+        --agora_benchmark agora_model_val
+
+PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
+srun -p ${PARTITION} \
+    --job-name=${JOB_NAME} \
+    --gres=gpu:${GPUS_PER_NODE} \
+    --ntasks-per-node=1 \
+    --cpus-per-task=${CPUS_PER_TASK} \
+    --quotatype=auto \
+    --kill-on-bad-exit=1 \
+    ${SRUN_ARGS} \
+    python test.py \
+        --num_gpus ${GPUS_PER_NODE} \
+        --exp_name output/test_${JOB_NAME}_ep${CKPT}_UBody \
+        --result_path ${RES_PATH} \
+        --ckpt_idx ${CKPT} \
+        --testset UBody \
+        --agora_benchmark agora_model_val
+        
+PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
+srun -p ${PARTITION} \
+    --job-name=${JOB_NAME} \
+    --gres=gpu:${GPUS_PER_NODE} \
+    --ntasks-per-node=1 \
+    --cpus-per-task=${CPUS_PER_TASK} \
+    --quotatype=auto \
+    --kill-on-bad-exit=1 \
+    ${SRUN_ARGS} \
+    python test.py \
+        --num_gpus ${GPUS_PER_NODE} \
+        --exp_name output/test_${JOB_NAME}_ep${CKPT}_EHF \
+        --result_path ${RES_PATH} \
+        --ckpt_idx ${CKPT} \
+        --testset EHF \
+        --agora_benchmark agora_model_val
+
+PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
+srun -p ${PARTITION} \
+    --job-name=${JOB_NAME} \
+    --gres=gpu:${GPUS_PER_NODE} \
+    --ntasks-per-node=1 \
+    --cpus-per-task=${CPUS_PER_TASK} \
+    --quotatype=auto \
+    --kill-on-bad-exit=1 \
+    ${SRUN_ARGS} \
+    python test.py \
+        --num_gpus ${GPUS_PER_NODE} \
+        --exp_name output/test_${JOB_NAME}_ep${CKPT}_AGORA_val \
+        --result_path ${RES_PATH} \
+        --ckpt_idx ${CKPT} \
+        --testset AGORA \
+        --agora_benchmark agora_model_val
+    
+PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
+srun -p ${PARTITION} \
+    --job-name=${JOB_NAME} \
+    --gres=gpu:${GPUS_PER_NODE} \
+    --ntasks-per-node=1 \
+    --cpus-per-task=${CPUS_PER_TASK} \
+    --quotatype=auto \
     --kill-on-bad-exit=1 \
-    --exclusive \
     ${SRUN_ARGS} \
     python test.py \
         --num_gpus ${GPUS_PER_NODE} \
@@ -93,69 +113,37 @@ srun -p ${PARTITION} \
         --result_path ${RES_PATH} \
         --ckpt_idx ${CKPT} \
         --testset ARCTIC \
-        # --agora_benchmark agora_model_val
+        --agora_benchmark agora_model_val
+w
+PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
+srun -p ${PARTITION} \
+    --job-name=${JOB_NAME} \
+    --gres=gpu:${GPUS_PER_NODE} \
+    --ntasks-per-node=1 \
+    --cpus-per-task=${CPUS_PER_TASK} \
+    --quotatype=auto \
+    --kill-on-bad-exit=1 \
+    ${SRUN_ARGS} \
+    python test.py \
+        --num_gpus ${GPUS_PER_NODE} \
+        --exp_name output/test_${JOB_NAME}_ep${CKPT}_RenBody \
+        --result_path ${RES_PATH} \
+        --ckpt_idx ${CKPT} \
+        --testset RenBody_HiRes \
+        --agora_benchmark agora_model_val
 
-# PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
-# srun -p ${PARTITION} \
-#     --job-name=${JOB_NAME} \
-#     --gres=gpu:${GPUS_PER_NODE} \
-#     --ntasks-per-node=1 \
-#     --cpus-per-task=${CPUS_PER_TASK} \
-#     --kill-on-bad-exit=1 \
-#     ${SRUN_ARGS} \
-#     python test.py \
-#         --num_gpus ${GPUS_PER_NODE} \
-#         --exp_name output/test_${JOB_NAME}_ep${CKPT}_RenBody_HiRes \
-#         --result_path ${RES_PATH} \
-#         --ckpt_idx ${CKPT} \
-#         --testset RenBody_HiRes \
-#         # --agora_benchmark agora_model_val
 
-# PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
-# srun -p ${PARTITION} \
-#     --job-name=${JOB_NAME} \
-#     --gres=gpu:${GPUS_PER_NODE} \
-#     --ntasks-per-node=1 \
-#     --cpus-per-task=${CPUS_PER_TASK} \
-#     --kill-on-bad-exit=1 \
-#     ${SRUN_ARGS} \
-#     python test.py \
-#         --num_gpus ${GPUS_PER_NODE} \
-#         --exp_name output/test_${JOB_NAME}_ep${CKPT}_EHF \
-#         --result_path ${RES_PATH} \
-#         --ckpt_idx ${CKPT} \
-#         --testset EHF \
-#         # --agora_benchmark agora_model_val
-
-# PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
-# srun -p ${PARTITION} \
-#     --job-name=${JOB_NAME} \
-#     --gres=gpu:${GPUS_PER_NODE} \
-#     --ntasks-per-node=1 \
-#     --cpus-per-task=${CPUS_PER_TASK} \
-#     --kill-on-bad-exit=1 \
-#     ${SRUN_ARGS} \
-#     python test.py \
-#         --num_gpus ${GPUS_PER_NODE} \
-#         --exp_name output/test_${JOB_NAME}_ep${CKPT}_AGORA_val \
-#         --result_path ${RES_PATH} \
-#         --ckpt_idx ${CKPT} \
-#         --testset AGORA \
-#         # --agora_benchmark agora_model_val
-
-# # PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
-# # srun -p ${PARTITION} \
-# #     --job-name=${JOB_NAME} \
-# #     --gres=gpu:${GPUS_PER_NODE} \
-# #     --ntasks-per-node=1 \
-# #     --cpus-per-task=${CPUS_PER_TASK} \
-# #     --kill-on-bad-exit=1 \
-# #     ${SRUN_ARGS} \
-# #     python test.py \
-# #         --num_gpus ${GPUS_PER_NODE} \
-# #         --exp_name output/test_${JOB_NAME} \
-# #         --result_path ${RES_PATH} \
-# #         --ckpt_idx ${CKPT} \
-# #         --testset AGORA \
-# #         --agora_benchmark test_only # agora-test with non-agora model; 'agora_model_test' for testing agora model
-# #         --agora_benchmark agora_model_test
+srun -p ${PARTITION} \
+    --job-name=${JOB_NAME} \
+    --gres=gpu:${GPUS_PER_NODE} \
+    --ntasks-per-node=1 \
+    --cpus-per-task=${CPUS_PER_TASK} \
+    --kill-on-bad-exit=1 \
+    ${SRUN_ARGS} \
+    python test.py \
+        --num_gpus ${GPUS_PER_NODE} \
+        --exp_name output/test_${JOB_NAME}_ep${CKPT}_AGORA_test \
+        --result_path ${RES_PATH} \
+        --ckpt_idx ${CKPT} \
+        --testset AGORA \
+        --agora_benchmark agora_model_test

main/slurm_test_wo_decoder.sh

@@ -1,153 +0,0 @@
-#!/usr/bin/env bash
-#!/usr/bin/env bash
-
-set -x
-
-PARTITION=Zoetrope
-JOB_NAME=$1
-GPUS=$2
-RES_PATH=$3
-CKPT=$4
-
-# model_path=../output/train_gta_synbody_ft_20230410_132110/model_dump/snapshot_2.pth.tar
-
-GPUS_PER_NODE=$((${GPUS}<8?${GPUS}:8))
-CPUS_PER_TASK=${CPUS_PER_TASK:-2}
-SRUN_ARGS=${SRUN_ARGS:-""}
-
-PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
-srun -p ${PARTITION} \
-    --job-name=${JOB_NAME} \
-    --gres=gpu:${GPUS_PER_NODE} \
-    --ntasks-per-node=1 \
-    --cpus-per-task=${CPUS_PER_TASK} \
-    --quotatype=auto \
-    --kill-on-bad-exit=1 \
-    ${SRUN_ARGS} \
-    python test.py \
-        --num_gpus ${GPUS_PER_NODE} \
-        --exp_name output/test_${JOB_NAME}_ep${CKPT}_PW3D \
-        --result_path ${RES_PATH} \
-        --ckpt_idx ${CKPT} \
-        --testset PW3D \
-        --agora_benchmark agora_model_val
-
-PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
-srun -p ${PARTITION} \
-    --job-name=${JOB_NAME} \
-    --gres=gpu:${GPUS_PER_NODE} \
-    --ntasks-per-node=1 \
-    --cpus-per-task=${CPUS_PER_TASK} \
-    --quotatype=auto \
-    --kill-on-bad-exit=1 \
-    ${SRUN_ARGS} \
-    python test.py \
-        --num_gpus ${GPUS_PER_NODE} \
-        --exp_name output/test_${JOB_NAME}_ep${CKPT}_EgoBody_Egocentric \
-        --result_path ${RES_PATH} \
-        --ckpt_idx ${CKPT} \
-        --testset EgoBody_Egocentric \
-        --agora_benchmark agora_model_val
-
-PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
-srun -p ${PARTITION} \
-    --job-name=${JOB_NAME} \
-    --gres=gpu:${GPUS_PER_NODE} \
-    --ntasks-per-node=1 \
-    --cpus-per-task=${CPUS_PER_TASK} \
-    --quotatype=auto \
-    --kill-on-bad-exit=1 \
-    ${SRUN_ARGS} \
-    python test.py \
-        --num_gpus ${GPUS_PER_NODE} \
-        --exp_name output/test_${JOB_NAME}_ep${CKPT}_UBody \
-        --result_path ${RES_PATH} \
-        --ckpt_idx ${CKPT} \
-        --testset UBody \
-        --agora_benchmark agora_model_val
-        
-PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
-srun -p ${PARTITION} \
-    --job-name=${JOB_NAME} \
-    --gres=gpu:${GPUS_PER_NODE} \
-    --ntasks-per-node=1 \
-    --cpus-per-task=${CPUS_PER_TASK} \
-    --quotatype=auto \
-    --kill-on-bad-exit=1 \
-    ${SRUN_ARGS} \
-    python test.py \
-        --num_gpus ${GPUS_PER_NODE} \
-        --exp_name output/test_${JOB_NAME}_ep${CKPT}_EHF \
-        --result_path ${RES_PATH} \
-        --ckpt_idx ${CKPT} \
-        --testset EHF \
-        --agora_benchmark agora_model_val
-
-PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
-srun -p ${PARTITION} \
-    --job-name=${JOB_NAME} \
-    --gres=gpu:${GPUS_PER_NODE} \
-    --ntasks-per-node=1 \
-    --cpus-per-task=${CPUS_PER_TASK} \
-    --quotatype=auto \
-    --kill-on-bad-exit=1 \
-    ${SRUN_ARGS} \
-    python test.py \
-        --num_gpus ${GPUS_PER_NODE} \
-        --exp_name output/test_${JOB_NAME}_ep${CKPT}_AGORA_val \
-        --result_path ${RES_PATH} \
-        --ckpt_idx ${CKPT} \
-        --testset AGORA \
-        --agora_benchmark agora_model_val
-    
-PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
-srun -p ${PARTITION} \
-    --job-name=${JOB_NAME} \
-    --gres=gpu:${GPUS_PER_NODE} \
-    --ntasks-per-node=1 \
-    --cpus-per-task=${CPUS_PER_TASK} \
-    --quotatype=auto \
-    --kill-on-bad-exit=1 \
-    ${SRUN_ARGS} \
-    python test.py \
-        --num_gpus ${GPUS_PER_NODE} \
-        --exp_name output/test_${JOB_NAME}_ep${CKPT}_ARCTIC \
-        --result_path ${RES_PATH} \
-        --ckpt_idx ${CKPT} \
-        --testset ARCTIC \
-        --agora_benchmark agora_model_val
-w
-PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
-srun -p ${PARTITION} \
-    --job-name=${JOB_NAME} \
-    --gres=gpu:${GPUS_PER_NODE} \
-    --ntasks-per-node=1 \
-    --cpus-per-task=${CPUS_PER_TASK} \
-    --quotatype=auto \
-    --kill-on-bad-exit=1 \
-    ${SRUN_ARGS} \
-    python test.py \
-        --num_gpus ${GPUS_PER_NODE} \
-        --exp_name output/test_${JOB_NAME}_ep${CKPT}_RenBody \
-        --result_path ${RES_PATH} \
-        --ckpt_idx ${CKPT} \
-        --testset RenBody_HiRes \
-        --agora_benchmark agora_model_val
-
-
-# srun -p ${PARTITION} \
-#     --job-name=${JOB_NAME} \
-#     --gres=gpu:${GPUS_PER_NODE} \
-#     --ntasks-per-node=1 \
-#     --cpus-per-task=${CPUS_PER_TASK} \
-#     --kill-on-bad-exit=1 \
-#     ${SRUN_ARGS} \
-#     python test.py \
-#         --num_gpus ${GPUS_PER_NODE} \
-#         --exp_name output/test_${JOB_NAME} \
-#         --result_path ${RES_PATH} \
-#         --ckpt_idx ${CKPT} \
-#         --testset AGORA \
-#         --agora_benchmark agora_model_test
-#         # --agora_benchmark test_only # agora-test with non-agora model; 'agora_model_test' for testing agora model
-         

main/slurm_train.sh

@@ -8,7 +8,7 @@ GPUS=$2
 CONFIG=$3
 
 GPUS_PER_NODE=$((${GPUS}<8?${GPUS}:8))
-CPUS_PER_TASK=4 # ${CPUS_PER_TASK:-2}
+CPUS_PER_TASK=4
 SRUN_ARGS=${SRUN_ARGS:-""}
 
 PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \