mirror of
https://github.com/hpcaitech/Open-Sora.git
synced 2026-04-10 21:01:26 +02:00
18 lines
581 B
Bash
18 lines
581 B
Bash
# !/bin/bash
|
|
|
|
VIDEO_DIR=$1
|
|
CKPT_DIR=$2
|
|
LOG_BASE=$CKPT_DIR
|
|
mkdir -p $LOG_BASE
|
|
echo "Logging to $LOG_BASE"
|
|
|
|
GPUS=(0 1 2 3 4 5 6 7)
|
|
START_INDEX_LIST=(0 2 4 6 8 10 12 14)
|
|
END_INDEX_LIST=(2 4 6 8 10 12 14 16)
|
|
TASK_ID_LIST=(calc_vbench_a calc_vbench_b calc_vbench_c calc_vbench_d calc_vbench_e calc_vbench_f calc_vbench_g calc_vbench_h) # for log records only
|
|
|
|
|
|
for i in "${!GPUS[@]}"; do
|
|
CUDA_VISIBLE_DEVICES=${GPUS[i]} python eval/vbench/calc_vbench.py $VIDEO_DIR $CKPT_DIR --start ${START_INDEX_LIST[i]} --end ${END_INDEX_LIST[i]} > ${LOG_BASE}/${TASK_ID_LIST[i]}.log 2>&1 &
|
|
done
|