diff --git a/src/otx/recipe/semantic_segmentation/dino_v2_tile.yaml b/src/otx/recipe/semantic_segmentation/dino_v2_tile.yaml index e645a193bc..4dc476089e 100644 --- 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