Skip to content

Commit 51d1adf

Browse files
authored
Fix names for ignored scope [HOT-FIX, 2.2.0] (#3924)
fix names for ignored scope
1 parent 7f1c7da commit 51d1adf

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/otx/algo/segmentation/litehrnet.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def _exporter(self) -> OTXModelExporter:
8787
@property
8888
def ignore_scope(self) -> dict[str, Any]:
8989
"""Get the ignored scope for LiteHRNet."""
90-
if self.model_version == "large":
90+
if self.model_version == "lite_hrnet_x":
9191
return {
9292
"ignored_scope": {
9393
"patterns": ["__module.model.decode_head.aggregator/*"],
@@ -175,7 +175,7 @@ def ignore_scope(self) -> dict[str, Any]:
175175
"preset": "performance",
176176
}
177177

178-
if self.model_version == "medium":
178+
if self.model_version == "lite_hrnet_18":
179179
return {
180180
"ignored_scope": {
181181
"patterns": ["__module.model.backbone/*"],
@@ -263,7 +263,7 @@ def ignore_scope(self) -> dict[str, Any]:
263263
"preset": "mixed",
264264
}
265265

266-
if self.model_version == "small":
266+
if self.model_version == "lite_hrnet_s":
267267
return {
268268
"ignored_scope": {
269269
"names": [

0 commit comments

Comments
 (0)