feat: ⚡ update topiq train scripts

chaofengc · Oct 5, 2023 · 81edc66 · 81edc66
1 parent c3b51ca
commit 81edc66
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/options/train/TOPIQ/train_TOPIQ_res50_koniq.yml b/options/train/TOPIQ/train_TOPIQ_res50_koniq.yml
@@ -1,4 +1,4 @@
-name: 001_CFANet_Res50_koniq
+name: 002_CFANet_Res50_koniq
 # name: debug_model
 model_type: GeneralIQAModel 
 num_gpu: 1  # set num_gpu: 0 for cpu mode
@@ -83,6 +83,10 @@ train:
     type: MSELoss 
     loss_weight: !!float 1.0
 
+  metric_loss_opt:
+    type: NiNLoss 
+    loss_weight: !!float 1.0
+
 # validation settings
 val:
   val_freq: !!float 800