This thread has been locked.

If you have a related question, please click the "Ask a related question" button in the top right corner. The newly created question will be automatically linked to this question.

TDA4VM: How to instruct TIDL to run particular layer with 16 bits accuracy

Part Number: TDA4VM

Hello,

I have trained my model with QAT, however to achieve acceptable accuracy I set few layers to be quantized with 16 bits accuracy by setting their bitwidth_weights and bitwidth_activations to 16.

After finishing the QAT, I export the onnx file to TIDL. The documentation instructs to set the advanced_options:output_feature_16bit_names_list with the layer names of the original model.

However the names are not exported into the ONNX correctly, they are swapped with node numbers, like in the figure below: I would like to set the Conv_9 to run at 16 bits, the layer name should appear at W field, but it is swapped with node 525. If I set 525 in the output_feature_16bit_names_list,

I receive error with non-existing layer name. How to overcome this issue?

Thank you,

Alex.