cancel
Showing results for 
Search instead for 
Did you mean: 

How to apply static quantization and pruning to my trained YOLO model using PyCharm?

困困de宇
Associate

When i used cubeai,i met some problem which i click anaylze the error is  "Unsupported layer types: DynamicQuantizeLinear, ConvInteger".And i saw my model's quantizelinear manner is dynamic.so i want used pycharm to solving my error.i want make a static quantization and pruning to my trained YOLO model.my model's type is PT file,can you give me a code to solve my problem and i will very appreciate for your help!4.png

1 REPLY 1
Julian E.
ST Employee

Hello @困困de宇,

 

You issue is that we don't currently support these layers.

Here is the list of supported layers for ONNX models: 

https://stedgeai-dc.st.com/assets/embedded-docs/supported_ops_onnx.html 

 

Concerning Yolo models, you can find architectures that we support here:

https://github.com/STMicroelectronics/stm32ai-modelzoo/tree/main/object_detection 

 

Have a good day,

Julian


In order to give better visibility on the answered topics, please click on 'Accept as Solution' on the reply which solved your issue or answered your question.