Fix wrong model name in converter & template (#4082)

* Fix wrong * Update CHAGELOG
openvinotoolkit · Oct 29, 2024 · 7bb36ef · 7bb36ef
1 parent a2d2c81
commit 7bb36ef
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 4 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -108,6 +108,8 @@ All notable changes to this project will be documented in this file.
   (<https://github.com/openvinotoolkit/training_extensions/pull/4057>)
 - Fix incorrect all_groups order configuration in HLabelInfo
   (<https://github.com/openvinotoolkit/training_extensions/pull/4067>)
+- Fix wrong model name in converter & template
+  (<https://github.com/openvinotoolkit/training_extensions/pull/4082>)
 
 ## \[v2.1.0\]
 

diff --git a/src/otx/__init__.py b/src/otx/__init__.py
@@ -3,7 +3,7 @@
 # Copyright (C) 2024 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-__version__ = "2.2.0rc12"
+__version__ = "2.2.0rc13"
 
 import os
 from pathlib import Path

diff --git a/src/otx/tools/converter.py b/src/otx/tools/converter.py
@@ -40,11 +40,11 @@
         "task": OTXTaskType.MULTI_CLASS_CLS,
         "model_name": "mobilenet_v3_large",
     },
-    "Custom_Image_Classification_EfficinetNet-B3": {
+    "Custom_Image_Classification_EfficientNet-B3": {
         "task": OTXTaskType.MULTI_CLASS_CLS,
         "model_name": "tv_efficientnet_b3",
     },
-    "Custom_Image_Classification_EfficinetNet-V2-L": {
+    "Custom_Image_Classification_EfficientNet-V2-L": {
         "task": OTXTaskType.MULTI_CLASS_CLS,
         "model_name": "tv_efficientnet_v2_l",
     },

diff --git a/src/otx/tools/templates/classification/efficientnet_b3/template.yaml b/src/otx/tools/templates/classification/efficientnet_b3/template.yaml
@@ -1,5 +1,5 @@
 # Description.
-model_template_id: Custom_Image_Classification_EfficinetNet-B3
+model_template_id: Custom_Image_Classification_EfficientNet-B3
 name: EfficientNet-B3
 task_type: CLASSIFICATION
 task_family: VISION