update dinov2 recipe

openvinotoolkit · Jul 16, 2024 · 68fbf5e · 68fbf5e
1 parent adf1f39
commit 68fbf5e
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/src/otx/recipe/classification/multi_class_cls/dino_v2.yaml b/src/otx/recipe/classification/multi_class_cls/dino_v2.yaml
@@ -1,22 +1,22 @@
 model:
-  class_path: otx.algo.classification.dino_v2.DINOv2RegisterClassifier
+  class_path: otx.algo.classification.vit.VisionTransformerForMulticlassCls
   init_args:
     label_info: 1000
-    freeze_backbone: False
-    backbone: dinov2_vits14_reg
+    arch: "dinov2-small"
 
     optimizer:
       class_path: torch.optim.AdamW
       init_args:
-        lr: 1e-5
+        lr: 0.0001
+        weight_decay: 0.05
 
     scheduler:
       class_path: lightning.pytorch.cli.ReduceLROnPlateau
       init_args:
-        mode: min
+        mode: max
         factor: 0.5
-        patience: 9
-        monitor: train/loss
+        patience: 1
+        monitor: val/accuracy
 
 engine:
   task: MULTI_CLASS_CLS