diff --git a/examples/export_ResNet18/resnet18.py b/examples/export_ResNet18/resnet18.py index 3ba1ff4eef95c9ce89b55fee27fb79bd5af5b7ca..935cf96a0779c90f55fa41cbf13e7571154fd642 100644 --- a/examples/export_ResNet18/resnet18.py +++ b/examples/export_ResNet18/resnet18.py @@ -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() # --------------------------------------------------------------