Fix config loss weights

mathpluscode · Jan 2, 2024 · c40b586 · c40b586
1 parent 277eee4
commit c40b586
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/imgx/conf/task/gaussian_diff_seg.yaml b/imgx/conf/task/gaussian_diff_seg.yaml
@@ -42,8 +42,8 @@ model:
 
 loss:
   dice: 1.0
-  cross_entropy: 0.0
-  focal: 1.0
+  cross_entropy: 1.0
+  focal: 0.0
   mse: 0.0
   vlb: 0.0
 

diff --git a/imgx/conf/task/seg.yaml b/imgx/conf/task/seg.yaml
@@ -16,8 +16,8 @@ model:
 
 loss:
   dice: 1.0
-  cross_entropy: 0.0
-  focal: 1.0
+  cross_entropy: 1.0
+  focal: 0.0
 
 early_stopping: # used on validation set
   metric: "mean_binary_dice_score_without_background"