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