I have a network of 3 layers built using Pytorch:
Input Dimension is [54, 576] and output Dim is [54, 4].
I export the model using torch.onnx.export() with flag export_params=True, then it is converted using snpe-onnx-to-dlc script.
python3 ./snpe-onnx-to-dlc ./reverse_last_branch.onnx --output_path ./reverse_last_branch.dlc --input_dim input "54,576" --out_name output
However for a same tensor of [54, 576] I am not able to get the same output before and after conversion to dlc format.
Do you have an idea why ?
Note that I have been able to get the same output for other models and each of those 3 layers return the exact same output after conversion when exported individually.