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

[Fix] Set datatype for passed input_tensor as well

parent 82561500
No related branches found
No related tags found
2 merge requests!710.4.0,!70[Fix] Get the input_tensor back to optional in the export() function
Pipeline #81221 passed
......@@ -79,6 +79,7 @@ def export(export_folder_name: str,
if inputs_tensor is not None:
if inputs_tensor.dtype() == aidge_core.dtype.int32:
set_nodes_datatypes(model) # Set datatype to int8 only
inputs_tensor.set_datatype(aidge_core.dtype.int8)
model.save(f"graph_log/3_set_datatypes")
# [aidge_cmp] Export feature maps tensors as json
......
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