diff --git a/examples/export_ResNet18/resnet18.py b/examples/export_ResNet18/resnet18.py index e39daf8df5dda8265eb95ee2b8e530882b2df0c8..f60e68ae1b7df4fcbeb1f52a0808b839f1ed5633 100644 --- a/examples/export_ResNet18/resnet18.py +++ b/examples/export_ResNet18/resnet18.py @@ -376,6 +376,8 @@ if (DO_EXAMPLES and quantize_model): print('\n MODEL ACCURACY = ', accuracy * 100, '%') print('\n QUANTIZED ACCURACY = ', quant_accuracy * 100, '%') + output_array = propagate(model, scheduler, aidge_tensors[0]) + if USE_CUDA: model.set_backend("cpu") for aidge_tensor in aidge_tensors: