search

quic / aimet

AIMET is a library that provides advanced quantization and compression techniques for trained neural network models.
https://quic.github.io/aimet-pages/index.html
Other
2.11k stars 376 forks source link

Channel pruning for SSD model supported? #1281

Closed krebin closed 2 weeks ago

krebin commented 2 years ago

Hello, I am trying to use AIMET to prune an SSD model in tensorflow. Upon looking at example code, I see that an evaluation function is required that measures accuracy. Is the API compatible with SSD architecture and is there an example I have not seen? Thanks

quic-ssiddego commented 2 years ago

Hi @krebin Could you please provide a snippet of what you have tried so far? And, also - have you checked this example? https://github.qualcomm.com/qualcomm-ai/aimet-model-zoo/blob/aimet_hf_transformer_model_definitions/zoo_tensorflow/Docs/SSDMobileNetV2.md