Skip to content
Snippets Groups Projects
Commit 6294bb9c authored by Axel Farrugia's avatar Axel Farrugia
Browse files

[Fix] Manually set the input after the quantzation step

parent 07dd7746
No related branches found
No related tags found
No related merge requests found
......@@ -140,7 +140,7 @@ DO_EXAMPLES = True
OPTIM_SIGN = False
SINGLE_SHIFT = True
ROUNDING = True
NO_QUANTIZATION = False
NO_QUANT = False
CLIPPING = aidge_quantization.Clipping.MSE # 'MAX'
FOLD_GRAPH = True
......@@ -345,7 +345,13 @@ Each time the graph has been change, it has to be reset.
Here some Quantizer and Cast nodes have been added.
"""
""" [Issue]
We need first to manually add an input tensor with the correct datatype,
as it is not automatically done in PTQ.
"""
if quantize_model:
input_node = model.get_ordered_inputs()[0]
input_node[0].get_operator().set_input(0, aidge_tensors[0])
scheduler.reset_scheduling()
# --------------------------------------------------------------
......
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