| ############### | |
| ## | |
| #### | |
| 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 | |