Add N_PATCHES env variable (#671)

kaiko-ai · Oct 9, 2024 · 4f1cbc7 · 4f1cbc7
1 parent 0e2cabe
commit 4f1cbc7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/configs/vision/pathology/offline/classification/camelyon16_small.yaml b/configs/vision/pathology/offline/classification/camelyon16_small.yaml
@@ -83,7 +83,7 @@ data:
           embeddings_transforms:
             class_path: eva.core.data.transforms.Pad2DTensor
             init_args:
-              pad_size: &N_PATCHES 500
+              pad_size: &N_PATCHES ${oc.env:N_PATCHES, 500}
           target_transforms:
             class_path: eva.core.data.transforms.dtype.ArrayToFloatTensor
       val:

diff --git a/configs/vision/pathology/offline/classification/panda_small.yaml b/configs/vision/pathology/offline/classification/panda_small.yaml
@@ -84,7 +84,7 @@ data:
           embeddings_transforms:
             class_path: eva.core.data.transforms.Pad2DTensor
             init_args:
-              pad_size: &N_PATCHES 200
+              pad_size: &N_PATCHES ${oc.env:N_PATCHES, 200}
       val:
         class_path: eva.datasets.MultiEmbeddingsClassificationDataset
         init_args: