Upload baseline configs

baseline/configs/4xDF2K_plksr_tiny.toml

@@ -0,0 +1,57 @@
+name = "4xDF2K_plksr_tiny"
+model_type = "image"
+scale = 4
+manual_seed = 1024
+
+[datasets.train]
+type = "paired"
+dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_HR_tiled/'
+dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_LR_tiled/'
+patch_size = 32
+batch_size = 16
+
+[datasets.val]
+name = "val"
+type = "paired"
+dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
+dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
+[val]
+val_freq = 10000
+save_img = false
+[val.metrics.psnr]
+type = "calculate_psnr"
+[val.metrics.ssim]
+type = "calculate_ssim"
+[val.metrics.dists]
+type = "calculate_dists"
+better = "lower"
+
+[path]
+
+[network_g]
+#type = "plksr"
+type = "plksr_tiny"
+
+[train]
+ema = 0.999
+
+[train.optim_g]
+type = "adamw"
+lr = 1e-4
+betas = [ 0.9, 0.99 ]
+weight_decay = 0
+
+[train.scheduler]
+type = "multisteplr"
+milestones = [ 60000, 120000 ]
+gamma = 0.5
+
+[train.pixel_opt]
+type = "L1Loss"
+loss_weight = 1.0
+reduction = "mean"
+
+[logger]
+total_iter = 1000000
+save_checkpoint_freq = 10000
+use_tb_logger = true

baseline/configs/4xDF2K_plksr_tiny_bf16.toml

@@ -0,0 +1,59 @@
+name = "4xDF2K_plksr_tiny_bf16"
+model_type = "image"
+scale = 4
+use_amp = true
+bfloat16 = true
+manual_seed = 1024
+
+[datasets.train]
+type = "paired"
+dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_HR_tiled/'
+dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_LR_tiled/'
+patch_size = 32
+batch_size = 16
+
+[datasets.val]
+name = "val"
+type = "paired"
+dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
+dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
+[val]
+val_freq = 10000
+save_img = false
+[val.metrics.psnr]
+type = "calculate_psnr"
+[val.metrics.ssim]
+type = "calculate_ssim"
+[val.metrics.dists]
+type = "calculate_dists"
+better = "lower"
+
+[path]
+
+[network_g]
+#type = "plksr"
+type = "plksr_tiny"
+
+[train]
+ema = 0.999
+
+[train.optim_g]
+type = "adamw"
+lr = 1e-4
+betas = [ 0.9, 0.99 ]
+weight_decay = 0
+
+[train.scheduler]
+type = "multisteplr"
+milestones = [ 60000, 120000 ]
+gamma = 0.5
+
+[train.pixel_opt]
+type = "L1Loss"
+loss_weight = 1.0
+reduction = "mean"
+
+[logger]
+total_iter = 200000
+save_checkpoint_freq = 10000
+use_tb_logger = true

baseline/configs/4xDF2K_plksr_tiny_fp16.toml

@@ -0,0 +1,58 @@
+name = "4xDF2K_plksr_tiny_fp16"
+model_type = "image"
+scale = 4
+use_amp = true
+manual_seed = 1024
+
+[datasets.train]
+type = "paired"
+dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_HR_tiled/'
+dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_LR_tiled/'
+patch_size = 32
+batch_size = 16
+
+[datasets.val]
+name = "val"
+type = "paired"
+dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
+dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
+[val]
+val_freq = 10000
+save_img = false
+[val.metrics.psnr]
+type = "calculate_psnr"
+[val.metrics.ssim]
+type = "calculate_ssim"
+[val.metrics.dists]
+type = "calculate_dists"
+better = "lower"
+
+[path]
+
+[network_g]
+#type = "plksr"
+type = "plksr_tiny"
+
+[train]
+ema = 0.999
+
+[train.optim_g]
+type = "adamw"
+lr = 1e-4
+betas = [ 0.9, 0.99 ]
+weight_decay = 0
+
+[train.scheduler]
+type = "multisteplr"
+milestones = [ 60000, 120000 ]
+gamma = 0.5
+
+[train.pixel_opt]
+type = "L1Loss"
+loss_weight = 1.0
+reduction = "mean"
+
+[logger]
+total_iter = 200000
+save_checkpoint_freq = 10000
+use_tb_logger = true