Update ssl configs

openvinotoolkit · Mar 13, 2023 · 24205e1 · 24205e1
1 parent 75ca427
commit 24205e1
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/otx/algorithms/classification/configs/efficientnet_b0_cls_incr/semisl/model_multilabel.py b/otx/algorithms/classification/configs/efficientnet_b0_cls_incr/semisl/model_multilabel.py
@@ -2,7 +2,7 @@
 
 # pylint: disable=invalid-name
 
-_base_ = "../../base/models/efficientnet.py"
+_base_ = ["../../../../../recipes/stages/classification/semisl.yaml", "../../base/models/efficientnet.py"]
 
 model = dict(
     task="classification",

diff --git a/otx/algorithms/classification/configs/efficientnet_v2_s_cls_incr/semisl/model_multilabel.py b/otx/algorithms/classification/configs/efficientnet_v2_s_cls_incr/semisl/model_multilabel.py
@@ -2,7 +2,7 @@
 
 # pylint: disable=invalid-name
 
-_base_ = "../../base/models/efficientnet_v2.py"
+_base_ = ["../../../../../recipes/stages/classification/semisl.yaml", "../../base/models/efficientnet_v2.py"]
 
 model = dict(
     task="classification",

diff --git a/...lgorithms/classification/configs/mobilenet_v3_large_1_cls_incr/semisl/model_multilabel.py b/...lgorithms/classification/configs/mobilenet_v3_large_1_cls_incr/semisl/model_multilabel.py
@@ -2,7 +2,7 @@
 
 # pylint: disable=invalid-name
 
-_base_ = "../../base/models/mobilenet_v3.py"
+_base_ = ["../../../../../recipes/stages/classification/semisl.yaml", "../../base/models/mobilenet_v3.py"]
 
 model = dict(
     task="classification",