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)