From 5eb439c72f4dc8584184dc3867c68f262e122a0f Mon Sep 17 00:00:00 2001 From: Nicolas Kaenzig Date: Fri, 6 Sep 2024 14:58:04 +0200 Subject: [PATCH] fixed number of classes in consep configs --- configs/vision/pathology/offline/segmentation/consep.yaml | 2 +- configs/vision/pathology/online/segmentation/consep.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/configs/vision/pathology/offline/segmentation/consep.yaml b/configs/vision/pathology/offline/segmentation/consep.yaml index 729aa9c4..2733670b 100644 --- a/configs/vision/pathology/offline/segmentation/consep.yaml +++ b/configs/vision/pathology/offline/segmentation/consep.yaml @@ -52,7 +52,7 @@ model: class_path: eva.vision.models.networks.decoders.segmentation.ConvDecoderMS init_args: in_features: ${oc.env:IN_FEATURES, 384} - num_classes: &NUM_CLASSES 8 + num_classes: &NUM_CLASSES 5 criterion: class_path: eva.vision.losses.DiceLoss init_args: diff --git a/configs/vision/pathology/online/segmentation/consep.yaml b/configs/vision/pathology/online/segmentation/consep.yaml index 63ea1eb4..85ec9b6a 100644 --- a/configs/vision/pathology/online/segmentation/consep.yaml +++ b/configs/vision/pathology/online/segmentation/consep.yaml @@ -45,7 +45,7 @@ model: class_path: eva.vision.models.networks.decoders.segmentation.ConvDecoderMS init_args: in_features: ${oc.env:IN_FEATURES, 384} - num_classes: &NUM_CLASSES 8 + num_classes: &NUM_CLASSES 5 criterion: class_path: eva.vision.losses.DiceLoss init_args: