Skip to content

Commit 2f041e5

Browse files
committed
all model configs add ema option
1 parent e57b757 commit 2f041e5

24 files changed

+24
-0
lines changed

options/train/BasicVSR/train_BasicVSR_REDS.yml

+1
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@ path:
5959

6060
# training settings
6161
train:
62+
ema_decay: 0.999
6263
optim_g:
6364
type: Adam
6465
lr: !!float 2e-4

options/train/BasicVSR/train_BasicVSR_Vimeo90K_BDx4.yml

+1
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@ path:
5858

5959
# training settings
6060
train:
61+
ema_decay: 0.999
6162
optim_g:
6263
type: Adam
6364
lr: !!float 2e-4

options/train/BasicVSR/train_BasicVSR_Vimeo90K_BIx4.yml

+1
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@ path:
5858

5959
# training settings
6060
train:
61+
ema_decay: 0.999
6162
optim_g:
6263
type: Adam
6364
lr: !!float 2e-4

options/train/BasicVSR/train_IconVSR_REDS.yml

+1
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ path:
6262

6363
# training settings
6464
train:
65+
ema_decay: 0.999
6566
optim_g:
6667
type: Adam
6768
lr: !!float 2e-4

options/train/BasicVSR/train_IconVSR_Vimeo90K_BDx4.yml

+1
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@ path:
6161

6262
# training settings
6363
train:
64+
ema_decay: 0.999
6465
optim_g:
6566
type: Adam
6667
lr: !!float 2e-4

options/train/BasicVSR/train_IconVSR_Vimeo90K_BIx4.yml

+1
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@ path:
6161

6262
# training settings
6363
train:
64+
ema_decay: 0.999
6465
optim_g:
6566
type: Adam
6667
lr: !!float 2e-4

options/train/EDSR/train_EDSR_Lx2.yml

+1
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ path:
6060

6161
# training settings
6262
train:
63+
ema_decay: 0.999
6364
optim_g:
6465
type: Adam
6566
lr: !!float 1e-4

options/train/EDSR/train_EDSR_Lx3.yml

+1
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ path:
6060

6161
# training settings
6262
train:
63+
ema_decay: 0.999
6364
optim_g:
6465
type: Adam
6566
lr: !!float 1e-4

options/train/EDSR/train_EDSR_Lx4.yml

+1
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ path:
6060

6161
# training settings
6262
train:
63+
ema_decay: 0.999
6364
optim_g:
6465
type: Adam
6566
lr: !!float 1e-4

options/train/EDSR/train_EDSR_Mx2.yml

+1
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ path:
6060

6161
# training settings
6262
train:
63+
ema_decay: 0.999
6364
optim_g:
6465
type: Adam
6566
lr: !!float 1e-4

options/train/EDSR/train_EDSR_Mx3.yml

+1
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ path:
6060

6161
# training settings
6262
train:
63+
ema_decay: 0.999
6364
optim_g:
6465
type: Adam
6566
lr: !!float 1e-4

options/train/EDSR/train_EDSR_Mx4.yml

+1
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ path:
6060

6161
# training settings
6262
train:
63+
ema_decay: 0.999
6364
optim_g:
6465
type: Adam
6566
lr: !!float 1e-4

options/train/EDVR/train_EDVRM_woTSA_GAN_TODO.yml

+1
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,7 @@ path:
7777

7878
# training settings
7979
train:
80+
ema_decay: 0.999
8081
optim_g:
8182
type: Adam
8283
lr: !!float 1e-4

options/train/EDVR/train_EDVR_L_x4_SR_REDS.yml

+1
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@ datasets:
4848

4949
# network structures
5050
network_g:
51+
ema_decay: 0.999
5152
type: EDVR
5253
num_in_ch: 3
5354
num_out_ch: 3

options/train/EDVR/train_EDVR_L_x4_SR_REDS_woTSA.yml

+1
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@ datasets:
4848

4949
# network structures
5050
network_g:
51+
ema_decay: 0.999
5152
type: EDVR
5253
num_in_ch: 3
5354
num_out_ch: 3

options/train/EDVR/train_EDVR_M_x4_SR_REDS.yml

+1
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@ path:
6969

7070
# training settings
7171
train:
72+
ema_decay: 0.999
7273
optim_g:
7374
type: Adam
7475
lr: !!float 4e-4

options/train/EDVR/train_EDVR_M_x4_SR_REDS_woTSA.yml

+1
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@ path:
6969

7070
# training settings
7171
train:
72+
ema_decay: 0.999
7273
optim_g:
7374
type: Adam
7475
lr: !!float 4e-4

options/train/ESRGAN/train_ESRGAN_x4.yml

+1
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@ path:
6161

6262
# training settings
6363
train:
64+
ema_decay: 0.999
6465
optim_g:
6566
type: Adam
6667
lr: !!float 1e-4

options/train/ESRGAN/train_RRDBNet_PSNR_x4.yml

+1
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,7 @@ path:
5757

5858
# training settings
5959
train:
60+
ema_decay: 0.999
6061
optim_g:
6162
type: Adam
6263
lr: !!float 2e-4

options/train/RCAN/train_RCAN_x2.yml

+1
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@ path:
6363

6464
# training settings
6565
train:
66+
ema_decay: 0.999
6667
optim_g:
6768
type: Adam
6869
lr: !!float 1e-4

options/train/SRResNet_SRGAN/train_MSRGAN_x4.yml

+1
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@ path:
6666

6767
# training settings
6868
train:
69+
ema_decay: 0.999
6970
optim_g:
7071
type: Adam
7172
lr: !!float 1e-4

options/train/SRResNet_SRGAN/train_MSRResNet_x2.yml

+1
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ path:
6060

6161
# training settings
6262
train:
63+
ema_decay: 0.999
6364
optim_g:
6465
type: Adam
6566
lr: !!float 2e-4

options/train/SRResNet_SRGAN/train_MSRResNet_x3.yml

+1
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ path:
6060

6161
# training settings
6262
train:
63+
ema_decay: 0.999
6364
optim_g:
6465
type: Adam
6566
lr: !!float 2e-4

options/train/SRResNet_SRGAN/train_MSRResNet_x4.yml

+1
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ path:
6060

6161
# training settings
6262
train:
63+
ema_decay: 0.999
6364
optim_g:
6465
type: Adam
6566
lr: !!float 2e-4

0 commit comments

Comments
 (0)