Update NAFNet-width32.yml

42 is a good number.
2022-04-12 22:38:12 +08:00 · 2022-04-12 22:38:12 +08:00 · 47ca917b2b
parent e9d02b0b03
commit 47ca917b2b
1 changed files with 2 additions and 2 deletions
--- a/options/train/GoPro/NAFNet-width32.yml
+++ b/options/train/GoPro/NAFNet-width32.yml
@ -9,7 +9,7 @@ name: NAFNet-GoPro-width32
 model_type: ImageRestorationModel
 scale: 1
 num_gpu: 8
-manual_seed: 10
+manual_seed: 42

 datasets:
  train:
@ -60,7 +60,7 @@ train:
  optim_g:
    type: AdamW
    lr: !!float 1e-3
-    weight_decay: 0.
+    weight_decay: !!float 1e-3
    betas: [0.9, 0.9]

  scheduler: