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)