train_espdnet_sun_rgbd.sh

CUDA_VISIBLE_DEVICES=0 python train_segmentation.py \
  --dataset sun \
  --data-path ./vision_datasets/sun_rgbd/ \
  --batch-size 35 \
  --crop-size 320 240 \
  --ignore-idx 255 \
  --model espdnet \
  --s 2.0 \
  --lr 0.010 \
  --scheduler hybrid \
  --clr-max 61 \
  --epochs 500

# CUDA_VISIBLE_DEVICES=0 python train_segmentation.py \
#   --dataset greenhouse \
#   --data-path ./vision_datasets/greenhouse/ \
#   --batch-size 30 \
#   --crop-size 480 256 \
#   --ignore-idx 4 \
#   --model espnetv2 \
#   --s 2.0 \
#   --lr 0.009 \
#   --scheduler hybrid \
#   --clr-max 61 \
#   --epochs 500