Multiple-shots / scripts /train_4gpu.sh
CUHKWilliam's picture
5
c70812a
raw
history blame contribute delete
722 Bytes
python -u -m torch.distributed.launch --nproc_per_node=4 --master_port=18024 \
./train.py --datapath "~/MY_MOT/TWL/data" \
--benchmark coco \
--fold 0 \
--bsz 1 \
--nworker 8 \
--backbone resnet101 \
--feature_extractor_path "~/MY_MOT/TWL/logistic_project/DCAMA/backbones/swin_base_patch4_window12_384_22kto1k.pth" \
--logpath "~/MY_MOT/TWL/logistic_project/DCAMA/log" \
--lr 1e-4 \
--nepoch 50 \
--load "~/MY_MOT/TWL/logistic_project/DCAMA/checkpoint/coco-20i/swin_fold2.pt" \
--nshot 3
# --load "/research/d6/rshr/xjgao/twl/logistic_project/DCAMA/checkpoint/coco-20i/resnet50_fold0.pt" \