Open-Sora/eval/loss/launch_single.sh
Zheng Zangwei (Alex Zheng) f1c6b8b88e open-sora v1.3 code upload (#786)
Co-authored-by: gxyes <gxynoz@gmail.com>
2025-02-20 16:50:24 +08:00

11 lines
315 B
Bash

#!/bin/bash
CMD_FILE_CONFIG="eval/loss/eval_loss.py configs/opensora-pro/misc/eval_loss.py"
PORTS=$1
CKPT_PATH=$2
VID_PATH=$3
# only evaluate for 360p, 102f
RESOLUTION=360p
torchrun --master-port ${PORTS} --nproc_per_node 1 $CMD_FILE_CONFIG --data-path $VID_PATH --ckpt-path $CKPT_PATH --resolution ${RESOLUTION}