[Bug] fix rdn and srcnn train configs (#1392)

* fix srcnn * fix rdn
open-mmlab · Nov 2, 2022 · 73ba837 · 73ba837
1 parent 424c48c
commit 73ba837
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 4 deletions.
diff --git a/configs/rdn/rdn_x2c64b16_1xb16-1000k_div2k.py b/configs/rdn/rdn_x2c64b16_1xb16-1000k_div2k.py
@@ -71,6 +71,7 @@
 
 train_dataloader = dict(
     num_workers=4,
+    batch_size=16,
     persistent_workers=False,
     sampler=dict(type='InfiniteSampler', shuffle=True),
     dataset=dict(
@@ -109,7 +110,7 @@
 optim_wrapper = dict(
     constructor='DefaultOptimWrapperConstructor',
     type='OptimWrapper',
-    optimizer=dict(type='Adam', lr=1e-4, betas=(0.9, 0.99)))
+    optimizer=dict(type='Adam', lr=1e-4, betas=(0.9, 0.999)))
 
 # learning policy
 param_scheduler = dict(

diff --git a/configs/rdn/rdn_x3c64b16_1xb16-1000k_div2k.py b/configs/rdn/rdn_x3c64b16_1xb16-1000k_div2k.py
@@ -71,6 +71,7 @@
 
 train_dataloader = dict(
     num_workers=4,
+    batch_size=16,
     persistent_workers=False,
     sampler=dict(type='InfiniteSampler', shuffle=True),
     dataset=dict(
@@ -109,7 +110,7 @@
 optim_wrapper = dict(
     constructor='DefaultOptimWrapperConstructor',
     type='OptimWrapper',
-    optimizer=dict(type='Adam', lr=1e-4, betas=(0.9, 0.99)))
+    optimizer=dict(type='Adam', lr=1e-4, betas=(0.9, 0.999)))
 
 # learning policy
 param_scheduler = dict(

diff --git a/configs/rdn/rdn_x4c64b16_1xb16-1000k_div2k.py b/configs/rdn/rdn_x4c64b16_1xb16-1000k_div2k.py
@@ -71,6 +71,7 @@
 
 train_dataloader = dict(
     num_workers=4,
+    batch_size=16,
     persistent_workers=False,
     sampler=dict(type='InfiniteSampler', shuffle=True),
     dataset=dict(
@@ -110,7 +111,7 @@
 optim_wrapper = dict(
     constructor='DefaultOptimWrapperConstructor',
     type='OptimWrapper',
-    optimizer=dict(type='Adam', lr=1e-4, betas=(0.9, 0.99)))
+    optimizer=dict(type='Adam', lr=1e-4, betas=(0.9, 0.999)))
 
 # learning policy
 param_scheduler = dict(

diff --git a/configs/srcnn/srcnn_x4k915_1xb16-1000k_div2k.py b/configs/srcnn/srcnn_x4k915_1xb16-1000k_div2k.py
@@ -71,6 +71,7 @@
 
 train_dataloader = dict(
     num_workers=4,
+    batch_size=16,
     persistent_workers=False,
     sampler=dict(type='InfiniteSampler', shuffle=True),
     dataset=dict(
@@ -109,7 +110,7 @@
 optim_wrapper = dict(
     constructor='DefaultOptimWrapperConstructor',
     type='OptimWrapper',
-    optimizer=dict(type='Adam', lr=2e-4, betas=(0.9, 0.99)))
+    optimizer=dict(type='Adam', lr=2e-4, betas=(0.9, 0.999)))
 
 # learning policy
 param_scheduler = dict(