stability config: 4x_NERVE_clean_gan.yml

docs/stability/clean/configs/4x_NERVE_clean_gan.yml

@@ -0,0 +1,98 @@
+name: 4x_NERVE_clean_gan
+scale: 4
+use_amp: true
+amp_bf16: true
+use_channels_last: true
+fast_matmul: false
+use_compile: false
+compile_mode: default
+num_gpu: auto
+manual_seed: 1024
+
+datasets:
+  train:
+    name: LUCID Train
+    type: pairedimagedataset
+    dataroot_gt: [datasets/train/lucid_hc512/hr]
+    dataroot_lq: [datasets/train/lucid_hc512/lr_x4]
+    lq_size: 96
+    use_hflip: true
+    use_rot: true
+    num_worker_per_gpu: 8
+    batch_size_per_gpu: 8
+    accum_iter: 1
+  val:
+    name: Urban100 Val
+    type: pairedimagedataset
+    dataroot_gt: [datasets/val/urban100/hr]
+    dataroot_lq: [datasets/val/urban100/lr_x4]
+    filename_tmpl: "{}x4"  # Urban100 LR files are named img_001x4.png
+
+network_g:
+  type: nerve
+
+network_d:
+  type: dunet
+
+path:
+  pretrain_network_g: experiments/4x_NERVE_clean_fidelity/models/net_g_ema_latest.safetensors
+  param_key_g: ~
+  strict_load_g: true
+  resume_state: ~
+
+train:
+  ema_decay: 0.999
+  ema_power: 0.75
+  grad_clip: false
+  grad_clip_max_norm: 100
+  optim_g:
+    type: AdamW
+    lr: !!float 1e-4
+    weight_decay: 0
+    betas: [0.9, 0.99]
+  optim_d:
+    type: AdamW
+    lr: !!float 1e-4
+    weight_decay: 0
+    betas: [0.9, 0.99]
+  scheduler:
+    type: MultiStepLR
+    milestones: [5000, 8000]
+    gamma: 0.5
+  total_iter: 10000
+  warmup_iter: -1
+  losses:
+    - type: mssimloss
+      loss_weight: 0.5
+    - type: perceptualloss
+      criterion: charbonnier
+      loss_weight: 0.01
+    - type: hsluvloss
+      criterion: charbonnier
+      loss_weight: 1.0
+    - type: ganloss
+      gan_type: vanilla
+      loss_weight: 0.1
+
+val:
+  val_enabled: true
+  val_freq: 2000
+  save_img: false
+  tile_size: 0
+  tile_overlap: 8
+  metrics_enabled: true
+  metrics:
+    psnr:
+      type: calculate_psnr
+      crop_border: 4
+      test_y_channel: true
+    ssim:
+      type: calculate_ssim
+      crop_border: 4
+      test_y_channel: true
+
+logger:
+  print_freq: 100
+  save_checkpoint_freq: 1000
+  save_checkpoint_format: safetensors
+  use_tb_logger: true