fix dinov2 tiling

openvinotoolkit · Dec 19, 2024 · a73e214 · a73e214
1 parent 21c88c9
commit a73e214
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/otx/recipe/semantic_segmentation/dino_v2_tile.yaml b/src/otx/recipe/semantic_segmentation/dino_v2_tile.yaml
@@ -2,7 +2,7 @@ model:
   class_path: otx.algo.segmentation.dino_v2_seg.DinoV2Seg
   init_args:
     label_info: 2
-    model_name: dinov2_vits14
+    model_name: dinov2-small-seg
 
     optimizer:
       class_path: torch.optim.AdamW
@@ -30,8 +30,8 @@ data: ../_base_/data/semantic_segmentation_tile.yaml
 overrides:
   data:
     input_size:
-      - 560
-      - 560
+      - 518
+      - 518
     train_subset:
       transforms:
         - class_path: otx.core.data.transform_libs.torchvision.Resize