test_2K.yml

name: CAMixerSR_x4_2K
model_type: TileModel
scale: 4
num_gpu: 1  # set num_gpu: 0 for cpu mode
manual_seed: 0
tile: 64

datasets:
  test_00:  # the 1st test dataset
    name: F2K
    type: PairedImageDataset
    dataroot_gt: /mnt/bn/mmlab-wangyan-srdata/Flickr2K/F2K/HR
    dataroot_lq: /mnt/bn/mmlab-wangyan-srdata/Flickr2K/F2K/LR
    io_backend:
      type: disk

  test_01:  # the 1st test dataset
    name: Test2K
    type: PairedImageDataset
    dataroot_gt: /mnt/bn/mmlab-wangyan-srdata/Benchmarks/Test8K/test2k/HR/X4
    dataroot_lq: /mnt/bn/mmlab-wangyan-srdata/Benchmarks/Test8K/test2k/LR/X4
    io_backend:
      type: disk

# network structures
network_g:
  type: CAMixerSR
  ratio: 0.5
  scale: 4

# path
path:
  pretrain_network_g: ~ #path to model
  strict_load_g: true
  param_key_g: 'params_ema' #~ #'params_ema' #'params' # only for MAN, for MAN-T and MAN-L, using ~ 

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

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