diff --git a/imgx/conf/task/gaussian_diff_seg.yaml b/imgx/conf/task/gaussian_diff_seg.yaml index f66b3b7..40abe29 100644 --- 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 index 0284b09..e9f9227 100644 --- 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"