Skip to main content
Associate
August 7, 2024
Question

sparse model

  • August 7, 2024
  • 1 reply
  • 2179 views

I have a large PyTorch model that I pruned using unstructured methods to decrease its size. The model size can only decrease if I filter out the zeros and save the model in a sparse format.

However, my problem is that ONNX doesn't support the sparse format. Does anyone know how I could prune a model, filter out the zeros to reduce the model size, and implement it on an evaluation board?

 

 

1 reply

fauvarque.daniel
ST Employee
August 26, 2024

Sparsity is not supported by STM32Cube.AI either.