updated lits yaml files

kaiko-ai · Oct 14, 2024 · ed72513 · ed72513
1 parent b71df67
commit ed72513
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 8 deletions.
diff --git a/configs/vision/radiology/offline/segmentation/lits.yaml b/configs/vision/radiology/offline/segmentation/lits.yaml
@@ -12,7 +12,7 @@ trainer:
           refresh_rate: ${oc.env:TQDM_REFRESH_RATE, 1}
       - class_path: eva.vision.callbacks.SemanticSegmentationLogger
         init_args:
-          log_every_n_steps: 1000
+          log_every_n_epochs: 1
           log_images: false
       - class_path: lightning.pytorch.callbacks.ModelCheckpoint
         init_args:
@@ -59,7 +59,7 @@ model:
     criterion:
       class_path: eva.core.losses.CrossEntropyLoss
       init_args:
-        weight: [0.01, 0.1, 1.5]
+        weight: [0.05, 0.1, 1.5]
     optimizer:
       class_path: torch.optim.AdamW
       init_args:
@@ -118,6 +118,7 @@ data:
               class_path: eva.vision.data.transforms.common.ResizeAndClamp
               init_args:
                 size: ${oc.env:RESIZE_DIM, 224}
+                clamp_range: [-1008, 822]
                 mean: &NORMALIZE_MEAN ${oc.env:NORMALIZE_MEAN, [0.485, 0.456, 0.406]} 
                 std: &NORMALIZE_STD ${oc.env:NORMALIZE_STD, [0.229, 0.224, 0.225]}
         - class_path: eva.vision.datasets.LiTS
@@ -136,6 +137,7 @@ data:
       val:
         batch_size: *BATCH_SIZE
         num_workers: *N_DATA_WORKERS
+        shuffle: true
       test:
         batch_size: *BATCH_SIZE
         num_workers: *N_DATA_WORKERS

diff --git a/configs/vision/radiology/offline/segmentation/lits_balanced.yaml b/configs/vision/radiology/offline/segmentation/lits_balanced.yaml
@@ -3,7 +3,7 @@ trainer:
   class_path: eva.Trainer
   init_args:
     n_runs: &N_RUNS ${oc.env:N_RUNS, 1}
-    default_root_dir: &OUTPUT_ROOT ${oc.env:OUTPUT_ROOT, logs/${oc.env:MODEL_NAME, vit_small_patch16_224_dino}/lits}
+    default_root_dir: &OUTPUT_ROOT ${oc.env:OUTPUT_ROOT, logs/${oc.env:MODEL_NAME, vit_small_patch16_224_dino}/lits_balanced}
     max_steps: &MAX_STEPS ${oc.env:MAX_STEPS, 500000}
     callbacks:
       - class_path: eva.callbacks.ConfigurationLogger
@@ -29,7 +29,7 @@ trainer:
           mode: *MONITOR_METRIC_MODE
       - class_path: eva.callbacks.SegmentationEmbeddingsWriter
         init_args:
-          output_dir: &DATASET_EMBEDDINGS_ROOT ${oc.env:EMBEDDINGS_ROOT, ./data/embeddings}/${oc.env:MODEL_NAME, vit_small_patch16_224_dino}/lits
+          output_dir: &DATASET_EMBEDDINGS_ROOT ${oc.env:EMBEDDINGS_ROOT, ./data/embeddings}/${oc.env:MODEL_NAME, vit_small_patch16_224_dino}/lits_balanced
           dataloader_idx_map:
             0: train
             1: val
@@ -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:
@@ -118,6 +118,7 @@ data:
               class_path: eva.vision.data.transforms.common.ResizeAndClamp
               init_args:
                 size: ${oc.env:RESIZE_DIM, 224}
+                clamp_range: [-1008, 822]
                 mean: &NORMALIZE_MEAN ${oc.env:NORMALIZE_MEAN, [0.485, 0.456, 0.406]} 
                 std: &NORMALIZE_STD ${oc.env:NORMALIZE_STD, [0.229, 0.224, 0.225]}
         - class_path: eva.vision.datasets.LiTSBalanced

diff --git a/configs/vision/radiology/online/segmentation/lits.yaml b/configs/vision/radiology/online/segmentation/lits.yaml
@@ -51,7 +51,7 @@ model:
     criterion:
       class_path: eva.core.losses.CrossEntropyLoss
       init_args:
-        weight: [0.01, 0.1, 1.5]
+        weight: [0.05, 0.1, 1.5]
     lr_multiplier_encoder: 0.0
     optimizer:
       class_path: torch.optim.AdamW
@@ -95,6 +95,7 @@ data:
             class_path: eva.vision.data.transforms.common.ResizeAndClamp
             init_args:
               size: ${oc.env:RESIZE_DIM, 224}  
+              clamp_range: [-1008, 822]
               mean: *NORMALIZE_MEAN
               std: *NORMALIZE_STD
       val:

diff --git a/configs/vision/radiology/online/segmentation/lits_balanced.yaml b/configs/vision/radiology/online/segmentation/lits_balanced.yaml
@@ -3,7 +3,7 @@ trainer:
   class_path: eva.Trainer
   init_args:
     n_runs: &N_RUNS ${oc.env:N_RUNS, 1}
-    default_root_dir: &OUTPUT_ROOT ${oc.env:OUTPUT_ROOT, logs/${oc.env:MODEL_NAME, vit_small_patch16_224_dino}/lits}
+    default_root_dir: &OUTPUT_ROOT ${oc.env:OUTPUT_ROOT, logs/${oc.env:MODEL_NAME, vit_small_patch16_224_dino}/lits_balanced}
     max_steps: &MAX_STEPS ${oc.env:MAX_STEPS, 500000}
     log_every_n_steps: 6
     callbacks:
@@ -95,6 +95,7 @@ data:
             class_path: eva.vision.data.transforms.common.ResizeAndClamp
             init_args:
               size: ${oc.env:RESIZE_DIM, 224}  
+              clamp_range: [-1008, 822]
               mean: *NORMALIZE_MEAN
               std: *NORMALIZE_STD
       val: