Hello @Chetan_Deshmukh ,
I can not convert your pt model into onnx model.
And, here is the same problem. You can try it.
Khadas VIM 3 NPU issues - Technical Support - Khadas Community
Although the problem has been solved, i still do not know what caused it.
If it still break down, please provide your onnx model and your whole train code.