diff --git a/src/otx/algo/segmentation/litehrnet.py b/src/otx/algo/segmentation/litehrnet.py index 11b20d421f7..73ade597fd8 100644 --- a/src/otx/algo/segmentation/litehrnet.py +++ b/src/otx/algo/segmentation/litehrnet.py @@ -87,7 +87,7 @@ def _exporter(self) -> OTXModelExporter: @property def ignore_scope(self) -> dict[str, Any]: """Get the ignored scope for LiteHRNet.""" - if self.model_version == "large": + if self.model_version == "lite_hrnet_x": return { "ignored_scope": { "patterns": ["__module.model.decode_head.aggregator/*"], @@ -175,7 +175,7 @@ def ignore_scope(self) -> dict[str, Any]: "preset": "performance", } - if self.model_version == "medium": + if self.model_version == "lite_hrnet_18": return { "ignored_scope": { "patterns": ["__module.model.backbone/*"], @@ -263,7 +263,7 @@ def ignore_scope(self) -> dict[str, Any]: "preset": "mixed", } - if self.model_version == "small": + if self.model_version == "lite_hrnet_s": return { "ignored_scope": { "names": [