update: simplify RDN and MPRNet

ryanxingql · Oct 3, 2024 · 79961cb · 79961cb
1 parent 9f5f8ee
commit 79961cb
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/options/test/MPRNet/DIV2K_LMDB_G1_latest.yml b/options/test/MPRNet/DIV2K_LMDB_G1_latest.yml
@@ -21,9 +21,9 @@ network_g:
   type: MPRNet
   io_channels: 3
   n_feat: 16
-  scale_unetfeats: 16
-  scale_orsnetfeats: 16
-  num_cab: 4
+  scale_unetfeats: 8
+  scale_orsnetfeats: 8
+  num_cab: 1
 
 # path
 path:

diff --git a/options/test/RDN/DIV2K_LMDB_G1_latest.yml b/options/test/RDN/DIV2K_LMDB_G1_latest.yml
@@ -24,7 +24,7 @@ network_g:
   mid_channels: 32
   num_blocks: 4
   num_layers: 4
-  channel_growth: 32
+  channel_growth: 16
 
 # path
 path:

diff --git a/options/train/MPRNet/DIV2K_LMDB_G1.yml b/options/train/MPRNet/DIV2K_LMDB_G1.yml
@@ -43,9 +43,9 @@ network_g:
   type: MPRNet
   io_channels: 3
   n_feat: 16
-  scale_unetfeats: 16
-  scale_orsnetfeats: 16
-  num_cab: 4
+  scale_unetfeats: 8
+  scale_orsnetfeats: 8
+  num_cab: 1
 
 # path
 path:

diff --git a/options/train/RDN/DIV2K_LMDB_G1.yml b/options/train/RDN/DIV2K_LMDB_G1.yml
@@ -46,7 +46,7 @@ network_g:
   mid_channels: 32
   num_blocks: 4
   num_layers: 4
-  channel_growth: 32
+  channel_growth: 16
 
 # path
 path: