From ad4e55f3526a78bcf7f62487f481bb28365f010f Mon Sep 17 00:00:00 2001
From: mayorx <chenliangyux@gmail.com>
Date: Fri, 15 Apr 2022 15:55:03 +0800
Subject: [PATCH] add GoPro width 32 test config.

---
 options/test/GoPro/NAFNet-width32.yml | 60 +++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)
 create mode 100644 options/test/GoPro/NAFNet-width32.yml

diff --git a/options/test/GoPro/NAFNet-width32.yml b/options/test/GoPro/NAFNet-width32.yml
new file mode 100644
index 0000000..c53208e
--- /dev/null
+++ b/options/test/GoPro/NAFNet-width32.yml
@@ -0,0 +1,60 @@
+# ------------------------------------------------------------------------
+# Copyright (c) 2022 megvii-model. All Rights Reserved.
+# ------------------------------------------------------------------------
+# Modified from BasicSR (https://github.com/xinntao/BasicSR)
+# Copyright 2018-2020 BasicSR Authors
+# ------------------------------------------------------------------------
+# general settings
+name: NAFNet-GoPro-width32-test
+model_type: ImageRestorationModel
+scale: 1
+num_gpu: 1 # set num_gpu: 0 for cpu mode
+manual_seed: 10
+
+# dataset and data loader settings
+datasets:
+
+  test:
+    name: gopro-test
+    type: PairedImageDataset
+
+    dataroot_gt: ./datasets/GoPro/test/target.lmdb
+    dataroot_lq: ./datasets/GoPro/test/input.lmdb
+
+    io_backend:
+      type: lmdb
+
+# network structures
+network_g:
+  type: NAFNetLocal
+  width: 32
+  enc_blk_nums: [1, 1, 1, 28]
+  middle_blk_num: 1
+  dec_blk_nums: [1, 1, 1, 1]
+
+# path
+path:
+  pretrain_network_g: experiments/pretrained_models/NAFNet-GoPro-width32.pth
+  strict_load_g: true
+  resume_state: ~
+
+# validation settings
+val:
+  save_img: false
+  grids: false
+
+
+  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
+
+# dist training settings
+dist_params:
+  backend: nccl
+  port: 29500