Tiramisu-Compiler / tiramisu_pytorch

Integration of Tiramisu (Compiler) into PyTorch
26 stars 1 forks source link

Exploiting unstructured sparsity via tiramisu_pytorch #1

Open Msabih opened 4 years ago

Msabih commented 4 years ago

Hi,

If I have a model that has a number of weights set to zero via unstructured pruning, its it possible to utilize that from tiramisu_pytorch ?

rbaghdadi commented 4 years ago

Hi @Msabih , at this stage we still do not support sparse DNNs in this integration. If you are interested in helping with this please let us know. @IHIaadj is leading the integration of Tiramisu in Pytorch.

In order to have full support of sparse DNNs in tiramisu_pytorch we need to:

Feel to reach out by email, we can setup a zoom call to give you more details.