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

[Fix](LeNet) Core dumped when dtype = fp32

parent 1ccec6ee
No related branches found
No related tags found
2 merge requests!710.4.0,!58Fix the aidge_cmp feature
Pipeline #77460 failed
......@@ -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
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