updated core config

4 months ago · 40f8ed3d5f
parent 77274f469a
commit 40f8ed3d5f
1 changed files with 5 additions and 10 deletions
--- a/ultralytics/engine/exporter.py
+++ b/ultralytics/engine/exporter.py
@ -1103,16 +1103,11 @@ class Exporter:
        # Configure MCT manually for specific layers
        bit_cfg = BitWidthConfig()
        bit_cfg.set_manual_activation_bit_width(
-            [
-                NodeNameScopeFilter("mul"),
-                NodeNameScopeFilter("sub"),
-                # NodeNameScopeFilter('sub_1'),
-                NodeNameScopeFilter("add_6"),
-                # NodeNameScopeFilter('add_7'),
-                # NodeNameScopeFilter('stack')
-            ],
-            16,
-        )
+        [NodeNameScopeFilter('mul'),
+        NodeNameScopeFilter('sub'),
+        NodeNameScopeFilter('add_6'),
+        NodeNameScopeFilter('cat_17')
+        ], 16)

        config = mct.core.CoreConfig(
            mixed_precision_config=mct.core.MixedPrecisionQuantizationConfig(num_of_images=10),