CUHKWilliam's picture
5
c70812a
raw
history blame
436 Bytes
python -u -m torch.distributed.launch --nnodes=1 --nproc_per_node=4 --node_rank=0 --master_port=16006 \
./train.py --datapath "../datasets" \
--benchmark coco \
--fold 0 \
--bsz 12 \
--nworker 8 \
--backbone swin \
--feature_extractor_path "../backbones/swin_base_patch4_window12_384.pth" \
--logpath "./logs" \
--lr 1e-3 \
--nepoch 500