Skip to content

Commit 54743ca

Browse files
authored
Fix MaskRCNN SwinT NNCF Accuracy Drop (#3685)
fix swin-t nncf kinda
1 parent 6d309ab commit 54743ca

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

src/otx/algo/instance_segmentation/maskrcnn.py

+16
Original file line numberDiff line numberDiff line change
@@ -820,3 +820,19 @@ def _build_model(self, num_classes: int) -> MaskRCNN:
820820
train_cfg=train_cfg,
821821
test_cfg=test_cfg,
822822
)
823+
824+
@property
825+
def _optimization_config(self) -> dict[str, Any]:
826+
"""PTQ config for MaskRCNN-SwinT."""
827+
return {
828+
"ignored_scope": {
829+
"types": [
830+
"Add",
831+
"MVN",
832+
"Divide",
833+
"Multiply",
834+
],
835+
"validate": False,
836+
},
837+
"preset": "mixed",
838+
}

0 commit comments

Comments
 (0)