uncommented line

kaiko-ai · Oct 14, 2024 · b448442 · b448442
1 parent ed72513
commit b448442
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: