diff --git a/src/otx/algorithms/classification/configs/deit_tiny/model.py b/src/otx/algorithms/classification/configs/deit_tiny/model.py index a70cb08623c..f9d0954fbf8 100644 --- 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) \ No newline at end of file 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 index 71605b835fe..e7c66326072 100644 --- 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) \ No newline at end of file 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 index 5bfbeddf253..a1638ed7075 100644 --- 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) \ No newline at end of file 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 index f1ade509561..2a81c2068ea 100644 --- 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")] +) \ No newline at end of file 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 index 75e41c21605..a6f3e6b70d3 100644 --- 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) \ No newline at end of file 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 index 25fe739990c..e7a0e51841f 100644 --- 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")] +) \ No newline at end of file diff --git a/src/otx/recipes/stages/classification/incremental.yaml b/src/otx/recipes/stages/classification/incremental.yaml index e4afb48a52f..43a38e588c2 100644 --- 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" + } +] \ No newline at end of file