Integrate model.py adaptive hook config --> recipe

openvinotoolkit · Aug 16, 2023 · c71cce7 · c71cce7
1 parent 52e62c5
commit c71cce7
Show file tree

Hide file tree

Showing 7 changed files with 14 additions and 18 deletions.
diff --git a/src/otx/algorithms/classification/configs/deit_tiny/model.py b/src/otx/algorithms/classification/configs/deit_tiny/model.py
@@ -17,6 +17,4 @@
 fp16 = dict(loss_scale=512.0)
 
 optimizer = dict(_delete_=True, type="AdamW", lr=0.01, weight_decay=0.05)
-optimizer_config = dict(_delete_=True)
-
-custom_hooks = [dict(type="AdaptiveRepeatDataHook", priority="ABOVE_NORMAL")]
+optimizer_config = dict(_delete_=True)
diff --git a/src/otx/algorithms/classification/configs/efficientnet_b0_cls_incr/model.py b/src/otx/algorithms/classification/configs/efficientnet_b0_cls_incr/model.py
@@ -22,6 +22,4 @@
     ),
 )
 
-fp16 = dict(loss_scale=512.0)
-
-custom_hooks = [dict(type="AdaptiveRepeatDataHook", priority="ABOVE_NORMAL")]
+fp16 = dict(loss_scale=512.0)
diff --git a/src/otx/algorithms/classification/configs/efficientnet_v2_s_cls_incr/model.py b/src/otx/algorithms/classification/configs/efficientnet_v2_s_cls_incr/model.py
@@ -16,6 +16,4 @@
     head=dict(type="CustomLinearClsHead", loss=dict(type="CrossEntropyLoss", loss_weight=1.0)),
 )
 
-fp16 = dict(loss_scale=512.0)
-
-custom_hooks = [dict(type="AdaptiveRepeatDataHook", priority="ABOVE_NORMAL")]
+fp16 = dict(loss_scale=512.0)
diff --git a/src/otx/algorithms/classification/configs/mobilenet_v3_large_075_cls_incr/model.py b/src/otx/algorithms/classification/configs/mobilenet_v3_large_075_cls_incr/model.py
@@ -18,6 +18,4 @@
         in_channels=720,
         hid_channels=1280,
     ),
-)
-
-custom_hooks = [dict(type="AdaptiveRepeatDataHook", priority="ABOVE_NORMAL")]
+)
diff --git a/src/otx/algorithms/classification/configs/mobilenet_v3_large_1_cls_incr/model.py b/src/otx/algorithms/classification/configs/mobilenet_v3_large_1_cls_incr/model.py
@@ -22,6 +22,4 @@
     ),
 )
 
-fp16 = dict(loss_scale=512.0)
-
-custom_hooks = [dict(type="AdaptiveRepeatDataHook", priority="ABOVE_NORMAL")]
+fp16 = dict(loss_scale=512.0)
diff --git a/src/otx/algorithms/classification/configs/mobilenet_v3_small_cls_incr/model.py b/src/otx/algorithms/classification/configs/mobilenet_v3_small_cls_incr/model.py
@@ -10,6 +10,4 @@
 model = dict(
     type="CustomImageClassifier",
     task="classification",
-)
-
-custom_hooks = [dict(type="AdaptiveRepeatDataHook", priority="ABOVE_NORMAL")]
+)
diff --git a/src/otx/recipes/stages/classification/incremental.yaml b/src/otx/recipes/stages/classification/incremental.yaml
@@ -50,3 +50,11 @@ adaptive_validation_interval:
   max_interval: 5
   enable_adaptive_interval_hook: True
   enable_eval_before_run: True
+
+
+custom_hooks: [
+  {
+    type:"AdaptiveRepeatDataHook", 
+    priority:"ABOVE_NORMAL"
+  }
+]