Skip to content
Snippets Groups Projects
Commit 71ea7485 authored by auphelia's avatar auphelia
Browse files

[Builder] infer data types after bit width minimization

parent 83973c38
No related branches found
No related tags found
No related merge requests found
......@@ -488,6 +488,8 @@ def step_minimize_bit_width(model: ModelWrapper, cfg: DataflowBuildConfig):
if cfg.minimize_bit_width:
model = model.transform(MinimizeWeightBitWidth())
model = model.transform(MinimizeAccumulatorWidth())
# make sure the changed datatypes are propagated through the network
model = model.transform(InferDataTypes())
return model
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment