search

lucaslie / torchprune

A research library for pytorch-based neural network pruning, compression, and more.
https://people.csail.mit.edu/lucasl/
MIT License
161 stars 24 forks source link

Generalization of the Repo #2

Open briliantnugraha opened 3 years ago

briliantnugraha commented 3 years ago

Hi, Thank you for your research, it is quite an interesting one. Btw, could I know, is this repo generalizable to other CNN model/is it already PnP to other model?

TIA :)

lucaslie commented 3 years ago

Hi, thank you for your interest!

I would recommend checking out the README here. It should contain all the info to add your own CNN models. :)