name: CUST_Base_Plus_x4
model_type: SRModel
scale: 4
num_gpu: 1  # set num_gpu: 0 for cpu mode
manual_seed: 0

datasets:
  test_1:  # the 1st test dataset
    name: Set5
    type: PairedImageDataset
    dataroot_gt: /data/SR/testset/Set5/HR
    dataroot_lq: /data/SR/testset/Set5/LR_bicubic/X4
    filename_tmpl: '{}x4'
    io_backend:
      type: disk
  
  test_4:  # the 2nd test dataset
    name: Urban100
    type: PairedImageDataset
    dataroot_gt: /data/SR/testset/urban100/HR
    dataroot_lq: /data/SR/testset/urban100/LR_bicubic/X4
    filename_tmpl: '{}x4'
    io_backend:
      type: disk

  test_3:  # the 1st test dataset
    name: B100
    type: PairedImageDataset
    dataroot_gt: /data/SR/testset/B100/HR
    dataroot_lq: /data/SR/testset/B100/LR_bicubic/X4
    filename_tmpl: '{}x4'
    io_backend:
      type: disk
  
  test_2:  # the 2nd test dataset
    name: Set14
    type: PairedImageDataset
    dataroot_gt: /data/SR/testset/Set14/HR
    dataroot_lq: /data/SR/testset/Set14/LR_bicubic/X4
    filename_tmpl: '{}x4'
    io_backend:
      type: disk

  
  test_5:  # the 1st test dataset
    name: Manga109
    type: PairedImageDataset
    dataroot_gt: /data/SR/testset/Manga109/HR
    dataroot_lq: /data/SR/testset/Manga109/LR_bicubic/X4
    filename_tmpl: '{}x4'
    io_backend:
      type: disk
  
  # test_6:
  #   name: DIV2K_val100
  #   type: PairedImageDataset
  #   dataroot_gt: datasets/DIV2K/DIV2K_Valid/HR
  #   dataroot_lq: datasets/DIV2K/DIV2K_Valid/LR_bicubic/X2
  #   filename_tmpl: '{}x2'
  #   io_backend:
  #     type: disk
  
# network structures
network_g:
  type: CUSTNet
  upscaling_factor: 4  # 네트워크 스케일 2 -> 4 변경
  dim : 40
  patch_size : [12,14,16,18,12,14,16,18,12,14,16,18]
  # patch_size : [18,18,18,18,18,18,18,18,18,18,18,18]
  window_size : 8
  group_size : 10


# path
path:
  pretrain_network_g: /workspace/LMLT/experiments/cust_plus_x4/models/net_g_467500.pth
  strict_load_g: true
  param_key_g: params

# validation settings
val:
  save_img: true
  suffix: ~  # add suffix to saved images, if None, use exp name

  metrics:
    psnr: # metric name, can be arbitrary
      type: calculate_psnr
      crop_border: 4
      test_y_channel: true
    ssim:
      type: calculate_ssim
      crop_border: 4
      test_y_channel: true