Spaces:
Runtime error
Runtime error
File size: 1,076 Bytes
f1df74a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
config="configs/inference/config_both.yaml"
ckpt='./checkpoints/motionctrl.pth'
condtype='both'
condtype='object_motion'
condtype='camera_motion'
cond_dir="examples/"
res_dir="./outputs/"
if [ ! -d $res_dir ]; then
mkdir -p $res_dir
fi
save_dir=$res_dir/$condtype'_seed'$seed
use_ddp=0
if [ $use_ddp == 0 ]; then
CUDA_VISIBLE_DEVICES=7 python 'main/evaluation/motionctrl_inference.py' \
--seed 1234 \
--ckpt_path $ckpt \
--base $config \
--savedir $save_dir \
--n_samples 5 \
--bs 1 --height 256 --width 256 \
--unconditional_guidance_scale 7.5 \
--ddim_steps 50 \
--ddim_eta 1.0 \
--condtype $condtype \
--cond_dir $cond_dir \
# --save_imgs
fi
if [ $use_ddp == 1 ]; then
python3 -m torch.distributed.launch \
--nproc_per_node=3 --nnodes=1 --master_port=23466 \
main/evaluation/ddp_wrapper.py \
--module 'inference' \
--seed 2000 \
--ckpt_path $ckpt \
--base $config \
--savedir $res_dir/$name \
--n_samples 3 \
--bs 1 --height 256 --width 256 \
--unconditional_guidance_scale 7.5 \
--ddim_steps 50 \
--ddim_eta 1.0 \
--condtype $condtype \
--cond_dir $cond_dir
fi |