Update train_DiffIRS2_GAN_x2_V2.yml

IFF-0303 · Sep 19, 2023 · 6867320 · 6867320
1 parent 5e9cd79
commit 6867320
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/DiffIR-RealSR/options/train_DiffIRS2_GAN_x2_V2.yml b/DiffIR-RealSR/options/train_DiffIRS2_GAN_x2_V2.yml
@@ -1,11 +1,10 @@
 # general settings
-name: train_DiffIRGANS2_x2
+name: train_DiffIRGANS2v2_x2
 model_type: DiffIRGANS2Model
 scale: 2
 num_gpu: auto  # auto: can infer from your visible devices automatically. official: 4 GPUs
 manual_seed: 0
 
-# ----------------- options for synthesizing training data in RealESRGANModel ----------------- #
 # USM the ground-truth
 l1_gt_usm: True
 percep_gt_usm: True
@@ -126,7 +125,6 @@ network_d:
 
 # path
 path:
-  # use the pre-trained Real-ESRNet model
   pretrain_network_S1: ./experiments/train_DiffIRS1_x2/models/net_g_latest.pth
   pretrain_network_g: ./experiments/train_DiffIRS2_x2/models/net_g_latest.pth
   param_key_g: params_ema