fixed number of classes in consep configs

kaiko-ai · Sep 6, 2024 · 5eb439c · 5eb439c
1 parent 067a0fa
commit 5eb439c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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: