update DeepLabV3 script

qjadud1994 · Jul 3, 2021 · d3e8a7b · d3e8a7b
1 parent 3c2cc85
commit d3e8a7b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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
+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