From d3e8a7b2d23b5efbf33ac168a71bfce756d531d2 Mon Sep 17 00:00:00 2001 From: beom Date: Sat, 3 Jul 2021 03:45:32 +0000 Subject: [PATCH] update DeepLabV3 script --- DeepLabV3Plus-Pytorch/run.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/DeepLabV3Plus-Pytorch/run.sh b/DeepLabV3Plus-Pytorch/run.sh index 28350c7..3f14eec 100755 --- a/DeepLabV3Plus-Pytorch/run.sh +++ b/DeepLabV3Plus-Pytorch/run.sh @@ -6,8 +6,8 @@ LR=0.05 # training with 2 GPUs -CUDA_VISLBLE_DEVICES=0,1 python main.py --data_root ${ROOT} --model ${MODEL} --gpu_id 0,1 --amp --total_itrs ${ITER} --batch_size ${BATCH} --lr ${LR} +CUDA_VISLBLE_DEVICES=0,1 python main.py --data_root ${ROOT} --model ${MODEL} --gpu_id 0,1 --amp --total_itrs ${ITER} --batch_size ${BATCH} --lr ${LR} --crop_val # evalutation with crf -CUDA_VISIBLE_DEVICES=0,1 python eval.py --gpu_id 0,1 --data_root ${ROOT} --model ${MODEL} --val_batch_size 16 --ckpt checkpoints/best_${MODEL}_voc_os16.pth \ No newline at end of file +CUDA_VISIBLE_DEVICES=0,1 python eval.py --gpu_id 0,1 --data_root ${ROOT} --model ${MODEL} --val_batch_size 16 --ckpt checkpoints/best_${MODEL}_voc_os16.pth --crop_val \ No newline at end of file