Skip to content

Commit

Permalink
Fix MaskRCNN SwinT NNCF Accuracy Drop (#3685)
Browse files Browse the repository at this point in the history
fix swin-t nncf kinda
  • Loading branch information
eugene123tw authored Jun 26, 2024
1 parent 6d309ab commit 54743ca
Showing 1 changed file with 16 additions and 0 deletions.
16 changes: 16 additions & 0 deletions src/otx/algo/instance_segmentation/maskrcnn.py
Original file line number Diff line number Diff line change
Expand Up @@ -820,3 +820,19 @@ def _build_model(self, num_classes: int) -> MaskRCNN:
train_cfg=train_cfg,
test_cfg=test_cfg,
)

@property
def _optimization_config(self) -> dict[str, Any]:
"""PTQ config for MaskRCNN-SwinT."""
return {
"ignored_scope": {
"types": [
"Add",
"MVN",
"Divide",
"Multiply",
],
"validate": False,
},
"preset": "mixed",
}

0 comments on commit 54743ca

Please sign in to comment.