From b44844209707a796d69880879c736a0acb5543a3 Mon Sep 17 00:00:00 2001
From: Nicolas Kaenzig <nkaenzig@gmail.com>
Date: Mon, 14 Oct 2024 13:11:32 +0200
Subject: [PATCH] uncommented line

---
 .../vision/radiology/offline/segmentation/lits_balanced.yaml  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/configs/vision/radiology/offline/segmentation/lits_balanced.yaml b/configs/vision/radiology/offline/segmentation/lits_balanced.yaml
index 95d2cb29..a0059e34 100644
--- a/configs/vision/radiology/offline/segmentation/lits_balanced.yaml
+++ b/configs/vision/radiology/offline/segmentation/lits_balanced.yaml
@@ -58,8 +58,8 @@ model:
         num_classes:  &NUM_CLASSES 3
     criterion:
       class_path: eva.core.losses.CrossEntropyLoss
-      # init_args:
-      #   weight: [0.05, 0.1, 1.5]
+      init_args:
+        weight: [0.05, 0.1, 1.5]
     optimizer:
       class_path: torch.optim.AdamW
       init_args: