cancel
Showing results for 
Search instead for 
Did you mean: 

Encoder only transformer on STM32

MK06
Visitor

I am trying to deploy encoder only transformer model on STM32 by converting pytorch model to onnx format first and then using STM toolchain (STM32 CubeAI). However, I am constantly getting this error when I analyzed the model:

TOOL ERROR: cannot reshape array of size 0 into shape ()

 

I tried to debug my model but couldn't find any issue in it. Any guidance will be appreciated.

 

Thank you.

0 REPLIES 0