Remove obsolete base model path in classificaiton templates

openvinotoolkit · Feb 23, 2023 · 6c06b9a · 6c06b9a
1 parent 57886a0
commit 6c06b9a
Show file tree

Hide file tree

Showing 6 changed files with 1 addition and 8 deletions.
diff --git a/otx/algorithms/classification/configs/efficientnet_b0_cls_incr/template.yaml b/otx/algorithms/classification/configs/efficientnet_b0_cls_incr/template.yaml
@@ -15,7 +15,6 @@ entrypoints:
   base: otx.algorithms.classification.tasks.ClassificationTrainTask
   openvino: otx.algorithms.classification.tasks.ClassificationOpenVINOTask
   nncf: otx.algorithms.classification.tasks.nncf.ClassificationNNCFTask
-base_model_path: ../../adapters/deep_object_reid/configs/efficientnet_b0/template_experimental.yaml
 
 # Capabilities.
 capabilities:

diff --git a/otx/algorithms/classification/configs/efficientnet_v2_s_cls_incr/template.yaml b/otx/algorithms/classification/configs/efficientnet_v2_s_cls_incr/template.yaml
@@ -15,7 +15,6 @@ entrypoints:
   base: otx.algorithms.classification.tasks.ClassificationTrainTask
   openvino: otx.algorithms.classification.tasks.ClassificationOpenVINOTask
   nncf: otx.algorithms.classification.tasks.nncf.ClassificationNNCFTask
-base_model_path: ../../adapters/deep_object_reid/configs/efficientnet_v2_s/template_experimental.yaml
 
 # Capabilities.
 capabilities:

diff --git a/...lgorithms/classification/configs/mobilenet_v3_large_075_cls_incr/template_experiment.yaml b/...lgorithms/classification/configs/mobilenet_v3_large_075_cls_incr/template_experiment.yaml
@@ -15,7 +15,6 @@ entrypoints:
   base: otx.algorithms.classification.tasks.ClassificationTrainTask
   openvino: otx.algorithms.classification.tasks.ClassificationOpenVINOTask
   nncf: otx.algorithms.classification.tasks.nncf.ClassificationNNCFTask
-base_model_path: ../../adapters/deep_object_reid/configs/mobilenet_v3_large_075/template_experimental.yaml
 
 # Capabilities.
 capabilities:

diff --git a/otx/algorithms/classification/configs/mobilenet_v3_large_1_cls_incr/template.yaml b/otx/algorithms/classification/configs/mobilenet_v3_large_1_cls_incr/template.yaml
@@ -15,7 +15,6 @@ entrypoints:
   base: otx.algorithms.classification.tasks.ClassificationTrainTask
   openvino: otx.algorithms.classification.tasks.ClassificationOpenVINOTask
   nncf: otx.algorithms.classification.tasks.nncf.ClassificationNNCFTask
-base_model_path: ../../adapters/deep_object_reid/configs/mobilenet_v3_large_1/template_experimental.yaml
 
 # Capabilities.
 capabilities:

diff --git a/otx/algorithms/classification/configs/mobilenet_v3_small_cls_incr/template_experiment.yaml b/otx/algorithms/classification/configs/mobilenet_v3_small_cls_incr/template_experiment.yaml
@@ -15,7 +15,6 @@ entrypoints:
   base: otx.algorithms.classification.tasks.ClassificationTrainTask
   openvino: otx.algorithms.classification.tasks.ClassificationOpenVINOTask
   nncf: otx.algorithms.classification.tasks.nncf.ClassificationNNCFTask
-base_model_path: ../../adapters/deep_object_reid/configs/mobilenet_v3_small/template_experimental.yaml
 
 # Capabilities.
 capabilities:

diff --git a/otx/mpa/modules/models/heads/semisl_multilabel_cls_head.py b/otx/mpa/modules/models/heads/semisl_multilabel_cls_head.py
@@ -126,9 +126,7 @@ def __init__(
         self.aux_mlp = generate_aux_mlp(aux_mlp, in_channels)
 
     def forward_train(self, x, gt_label):
-        return self.forward_train_with_last_layers(
-            x, gt_label, final_cls_layer=self.fc, final_emb_layer=self.aux_mlp
-        )
+        return self.forward_train_with_last_layers(x, gt_label, final_cls_layer=self.fc, final_emb_layer=self.aux_mlp)
 
 
 @HEADS.register_module()