ReBac/run_disjoint.sh at master · ANDYZAQ/ReBac · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
MODEL=deeplabv3bga_resnet101
DATA_ROOT=/data/anqi/VOCdevkit/VOC2012
DATASET=voc
TASK=15-5
EPOCH=20
BATCH=8
LOSS=bce_loss
LR=0.001
THRESH=0.7
MEMORY=0

SUBPATH=BARM_fpsd

CURR=1

now=$(date +"%Y%m%d_%H%M%S")
result_dir=./checkpoints/${SUBPATH}/${TASK}/

if [ ! -d ${result_dir} ]; then
    mkdir -p ${result_dir}
fi

# CUDA_VISIBLE_DEVICES=6 \
# python train.py --data_root ${DATA_ROOT} --model ${MODEL} --crop_val --lr ${LR} \
#     --batch_size ${BATCH} --train_epoch ${EPOCH}  --loss_type ${LOSS} \
#     --dataset ${DATASET} --task ${TASK} --lr_policy poly \
#     --pseudo --pseudo_thresh ${THRESH}  --bn_freeze  --amp \
#     --curr_step ${CURR} --subpath ${SUBPATH} \
#     --overlap \
#     | tee ${result_dir}/train-$now.log

CUDA_VISIBLE_DEVICES=8,9 \
torchrun --nproc_per_node=2 --master_port=24124 \
train.py --data_root ${DATA_ROOT} --model ${MODEL} --crop_val --lr ${LR} \
    --batch_size ${BATCH} --train_epoch ${EPOCH}  --loss_type ${LOSS} \
    --dataset ${DATASET} --task ${TASK} --lr_policy poly \
    --pseudo --pseudo_thresh ${THRESH}  --bn_freeze  --amp\
    --curr_step ${CURR} --subpath ${SUBPATH} \
    | tee ${result_dir}/train-$now.log