###############
## 
####

GPU: [0]   # 当前环境只可见 1 张 GPU；多卡时改成 [0,1,...] 并同步 torchrun --nproc_per_node

VERBOSE: True

MODEL:
  MODE: 'Deraining'
  SESSION: 'MPRNet'

# Optimization arguments.
OPTIM:
  BATCH_SIZE: 16   # TRAIN_PS=192 时建议 8~12；若 OOM 改为 8 或 6
  NUM_EPOCHS: 200
  # NEPOCH_DECAY: [10]
  LR_INITIAL: 2e-4   # 初始学习率
  LR_MIN: 1e-6       # 最后一个 epoch 时的学习率（余弦衰减）
  # BETA1: 0.9

TRAINING:
  VAL_AFTER_EVERY: 1
  RESUME: False
  TRAIN_PS: 128   # 调大 patch；192 约 2.25x 显存，batch 建议 8~12；256 则用 batch 4~6
  VAL_PS: 160     # 验证 patch 可略小或与 TRAIN_PS 一致
  TRAIN_DIR: '/media/home/songmeixi_insta360.com/Low_light_rainy_new/dataset/train/syn+real'       # path to training data
  VAL_DIR: '/media/home/songmeixi_insta360.com/Low_light_rainy_new/dataset/train/syn+real' # path to validation data
  SAVE_DIR: './checkpoint_smoke'           # path to save models and images
  # SAVE_IMAGES: False