| name = "4x_figsr" | |
| model_type = "image" | |
| scale = 4 | |
| use_amp = true | |
| bfloat16 = false | |
| fast_matmul = false | |
| #compile = true | |
| manual_seed = 1024 | |
| [datasets.train] | |
| type = "paired" | |
| dataroot_gt = 'datasets/BHI' | |
| dataroot_lq = 'datasets/BHI_lq' | |
| patch_size = 64 | |
| batch_size = 64 | |
| #accumulate = 1 | |
| [datasets.val] | |
| name = "val" | |
| type = "paired" | |
| dataroot_gt = 'C:\datasets\val\gt\' | |
| dataroot_lq = 'C:\datasets\val\lq\' | |
| [val] | |
| val_freq = 5000 | |
| #tile = 200 | |
| [val.metrics.psnr] | |
| type = "calculate_psnr" | |
| [val.metrics.ssim] | |
| type = "calculate_ssim" | |
| #[val.metrics.dists] | |
| #type = "calculate_dists" | |
| #better = "lower" | |
| #[val.metrics.topiq] | |
| #type = "calculate_topiq" | |
| [path] | |
| #pretrain_network_g = 'experiments\pretrain_g.pth' | |
| #pretrain_network_d = 'experiments\pretrain_d.pth' | |
| [network_g] | |
| type = "FIGSR" | |
| [train] | |
| grad_clip = false | |
| ema = 0.999 | |
| wavelet_guided = false | |
| #wavelet_init = 80000 | |
| #sam = "fsam" | |
| #sam_init = 1000 | |
| #eco = true | |
| #eco_init = 15000 | |
| #match_lq_colors = true | |
| [train.optim_g] | |
| type = "adamw" | |
| lr = 5e-4 | |
| betas = [0.9, 0.99] | |
| weight_decay = 0.01 | |
| [train.scheduler] | |
| type = "multisteplr" | |
| milestones = [200000,400000,600000,800000] | |
| gamma = 0.5 | |
| # losses | |
| [train.mssim_opt] | |
| type = "huber_loss" | |
| loss_weight = 1.0 | |
| [logger] | |
| total_iter = 1000000 | |
| save_checkpoint_freq = 5000 | |
| use_tb_logger = true | |
| #save_tb_img = true | |
| #print_freq = 100 |