Skip to content

Commit 3941190

Browse files
authored
Update classification in converter (#4146)
1 parent d9a3cd8 commit 3941190

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/otx/tools/converter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -240,7 +240,7 @@ def convert(config_path: str, task: OTXTaskType | None = None) -> dict:
240240
if param_dict.get("enable_tiling", None) and not task_info["model_name"].endswith("_tile"):
241241
task_info["model_name"] += "_tile"
242242
# classification task type can't be deducted from template name, try to extract from config
243-
if "sub_task_type" in template_config and "CLS" in template_config["sub_task_type"]:
243+
if "sub_task_type" in template_config and "_CLS" in task_info["task"]:
244244
task_info["task"] = template_config["sub_task_type"]
245245
if task is not None:
246246
task_info["task"] = task

0 commit comments

Comments
 (0)