elia / scripts /test_lavt.py
yxchng
add files
a166479
raw
history blame
1.49 kB
import os
import sys
refcoco_ckpt_path = "/cluster/nvme4/cyx/lavt/vis/refcoco.pth"
refcocop_ckpt_path = "/cluster/nvme4/cyx/lavt/vis/refcoco+.pth"
cmd1 = 'python test_lavt.py --model="lavt" --swin_type="base" --dataset="refcoco" --split="val" --resume="{:s}" --workers="4" --ddp_trained_weights --window12 --img_size="480"'.format(refcoco_ckpt_path)
cmd2 = 'python test_lavt.py --model="lavt" --swin_type="base" --dataset="refcoco" --split="testA" --resume="{:s}" --workers="4" --ddp_trained_weights --window12 --img_size="480"'.format(refcoco_ckpt_path)
cmd3 = 'python test_lavt.py --model="lavt" --swin_type="base" --dataset="refcoco" --split="testB" --resume="{:s}" --workers="4" --ddp_trained_weights --window12 --img_size="480"'.format(refcoco_ckpt_path)
cmd4 = 'python test_lavt.py --model="lavt" --swin_type="base" --dataset="refcoco+" --split="val" --resume="{:s}" --workers="4" --ddp_trained_weights --window12 --img_size="480"'.format(refcocop_ckpt_path)
cmd5 = 'python test_lavt.py --model="lavt" --swin_type="base" --dataset="refcoco+" --split="testA" --resume="{:s}" --workers="4" --ddp_trained_weights --window12 --img_size="480"'.format(refcocop_ckpt_path)
cmd6 = 'python test_lavt.py --model="lavt" --swin_type="base" --dataset="refcoco+" --split="testB" --resume="{:s}" --workers="4" --ddp_trained_weights --window12 --img_size="480"'.format(refcocop_ckpt_path)
cmds = [cmd1,cmd2,cmd3,cmd4,cmd5,cmd6]
for cmd in cmds:
print(cmd, flush=True)
os.system(cmd)