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

[Chore](R18) Perform an inference to set the intermediate tensors to export

parent 726d6a3d
No related branches found
No related tags found
No related merge requests found
......@@ -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:
......
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