Auto-format by https://ultralytics.com/actions

4 months ago · 586dfad80e
parent 40f8ed3d5f
commit 586dfad80e
1 changed files with 8 additions and 5 deletions
--- a/ultralytics/engine/exporter.py
+++ b/ultralytics/engine/exporter.py
@ -1103,11 +1103,14 @@ class Exporter:
        # Configure MCT manually for specific layers
        bit_cfg = BitWidthConfig()
        bit_cfg.set_manual_activation_bit_width(
-        [NodeNameScopeFilter('mul'),
-        NodeNameScopeFilter('sub'),
-        NodeNameScopeFilter('add_6'),
-        NodeNameScopeFilter('cat_17')
-        ], 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),