From e477a1d6040fba02ee2517e3cebbaca5a698f399 Mon Sep 17 00:00:00 2001 From: Z-Fran <1396925302@qq.com> Date: Wed, 26 Oct 2022 16:16:27 +0800 Subject: [PATCH 1/2] fix srcnn --- configs/srcnn/srcnn_x4k915_1xb16-1000k_div2k.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/configs/srcnn/srcnn_x4k915_1xb16-1000k_div2k.py b/configs/srcnn/srcnn_x4k915_1xb16-1000k_div2k.py index e2032424a5..cf39811509 100644 --- 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( From 98a51f34c5ca0f90acd15f5b7847425b2e92458a Mon Sep 17 00:00:00 2001 From: Z-Fran <1396925302@qq.com> Date: Wed, 16 Nov 2022 11:48:06 +0800 Subject: [PATCH 2/2] [Fix] fix srgan train config --- .../srgan_resnet/msrresnet_x4c64b16_1xb16-1000k_div2k.py | 8 ++++---- configs/srgan_resnet/srgan_x4c64b16_1xb16-1000k_div2k.py | 3 +++ 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/configs/srgan_resnet/msrresnet_x4c64b16_1xb16-1000k_div2k.py b/configs/srgan_resnet/msrresnet_x4c64b16_1xb16-1000k_div2k.py index 4440d967d4..121a5e7390 100644 --- a/configs/srgan_resnet/msrresnet_x4c64b16_1xb16-1000k_div2k.py +++ b/configs/srgan_resnet/msrresnet_x4c64b16_1xb16-1000k_div2k.py @@ -89,15 +89,15 @@ sampler=dict(type='DefaultSampler', shuffle=False), dataset=dict( type=dataset_type, - metainfo=dict(dataset_type='set5', task_name='sisr'), - data_root=data_root + '/Set5', + metainfo=dict(dataset_type='set14', task_name='sisr'), + data_root=data_root + '/Set14', data_prefix=dict(img='LRbicx4', gt='GTmod12'), pipeline=val_pipeline)) val_evaluator = [ dict(type='MAE'), - dict(type='PSNR'), - dict(type='SSIM'), + dict(type='PSNR', crop_border=scale), + dict(type='SSIM', crop_border=scale), ] train_cfg = dict( diff --git a/configs/srgan_resnet/srgan_x4c64b16_1xb16-1000k_div2k.py b/configs/srgan_resnet/srgan_x4c64b16_1xb16-1000k_div2k.py index 090ca64826..a5f0c275e6 100644 --- a/configs/srgan_resnet/srgan_x4c64b16_1xb16-1000k_div2k.py +++ b/configs/srgan_resnet/srgan_x4c64b16_1xb16-1000k_div2k.py @@ -62,3 +62,6 @@ by_epoch=False, milestones=[50000, 100000, 200000, 300000], gamma=0.5) + +train_cfg = dict( + type='IterBasedTrainLoop', max_iters=400_000, val_interval=5000)