updated core config

mct-2.1.1
Francesco Mattioli 4 months ago
parent 77274f469a
commit 40f8ed3d5f
  1. 15
      ultralytics/engine/exporter.py

@ -1103,16 +1103,11 @@ class Exporter:
# Configure MCT manually for specific layers # Configure MCT manually for specific layers
bit_cfg = BitWidthConfig() bit_cfg = BitWidthConfig()
bit_cfg.set_manual_activation_bit_width( bit_cfg.set_manual_activation_bit_width(
[ [NodeNameScopeFilter('mul'),
NodeNameScopeFilter("mul"), NodeNameScopeFilter('sub'),
NodeNameScopeFilter("sub"), NodeNameScopeFilter('add_6'),
# NodeNameScopeFilter('sub_1'), NodeNameScopeFilter('cat_17')
NodeNameScopeFilter("add_6"), ], 16)
# NodeNameScopeFilter('add_7'),
# NodeNameScopeFilter('stack')
],
16,
)
config = mct.core.CoreConfig( config = mct.core.CoreConfig(
mixed_precision_config=mct.core.MixedPrecisionQuantizationConfig(num_of_images=10), mixed_precision_config=mct.core.MixedPrecisionQuantizationConfig(num_of_images=10),

Loading…
Cancel
Save