extract_smplx_pose.sh

#!/bin/bash

export TZ=Asia/Shanghai

LOG_TIME=$(date +%Y%m%d_%H%M%S)
CUR_DIR="/mnt/shared-storage-user/mllm/pmx/SMPL-X_pose_extraction"
cd $CUR_DIR

nohup stdbuf -oL bash -c "while true; do date; nvidia-smi; sleep 10; done" \
    > "${CUR_DIR}/extract_smplx_${LOG_TIME}_gpu_monitor.log" 2>&1 &

PY310_BIN="/mnt/shared-storage-user/mllm/pmx/envs/py310/bin/python3.10"
export PYTHONPATH="${CUR_DIR}:${CUR_DIR}/SMPLest-X:${CUR_DIR}/WiLoR:${CUR_DIR}/Depth-Anything-V2:$PYTHONPATH"
export OPENCV_FOR_THREADS_NUM=8

$PY310_BIN -u -m accelerate.commands.launch \
    --same_network \
    --num_processes 4 \
    --num_machines 1 \
    --mixed_precision bf16 \
    --dynamo_backend no \
    --num_cpu_threads_per_process 24 \
    extract_smplx_pose.py \
    --batch_size 147456 \
    --num_workers 64 \
    --input_path "/mnt/shared-storage-user/" \
    --output_path "/mnt/shared-storage-user/"