File size: 1,487 Bytes
a166479 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
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)
|