From ca60c68127ef4faefc0a1540cd0afdfab1ec6207 Mon Sep 17 00:00:00 2001
From: Axel Farrugia <axel.farrugia@cea.fr>
Date: Wed, 2 Jul 2025 09:47:33 +0200
Subject: [PATCH] [Fix](LeNet) Core dumped when dtype = fp32
---
examples/export_LeNet/lenet.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/examples/export_LeNet/lenet.py b/examples/export_LeNet/lenet.py
index d80a307..28c7517 100644
--- a/examples/export_LeNet/lenet.py
+++ b/examples/export_LeNet/lenet.py
@@ -306,8 +306,9 @@ if quantize_model:
tensors[i] = aidge_core.Tensor(array)
tensors[i].set_datatype(TARGET_TYPE)
tensors[i].set_backend("cpu")
- # Setting modele to CPU for export
- model.set_backend("cpu")
+
+# Setting model to CPU for export
+model.set_backend("cpu")
# --------------------------------------------------------------
@@ -479,4 +480,4 @@ try:
for std_line in aidge_core.utils.run_command(["./bin/run_export"], cwd=EXPORT_FOLDER):
print(std_line, end="")
except subprocess.CalledProcessError as e:
- raise RuntimeError(0, f"An error occurred, failed to run export.") from e
\ No newline at end of file
+ raise RuntimeError(0, f"An error occurred, failed to run export.") from e
--
GitLab