From be873655f9bc573d3e17c89f63f5d6804e83515e Mon Sep 17 00:00:00 2001
From: Axel Farrugia <axel.farrugia@cea.fr>
Date: Fri, 2 May 2025 13:49:18 +0200
Subject: [PATCH] [Chore](R18) Perform an inference to set the intermediate
tensors to export
---
examples/export_ResNet18/resnet18.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/examples/export_ResNet18/resnet18.py b/examples/export_ResNet18/resnet18.py
index e39daf8..f60e68a 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:
--
GitLab