search

kaiqzhan / admm-pruning

Prune DNN using Alternating Direction Method of Multipliers (ADMM)
BSD 2-Clause "Simplified" License
107 stars 34 forks source link

PyTorch Implementation #3

Open haimengzhao opened 6 years ago

haimengzhao commented 6 years ago

Is a PyTorch implementation going to be provided?

kaiqzhan commented 6 years ago

I’m interested to code a pytorch version too, but it’s not guaranteed.

bzantium commented 5 years ago

I implemented pytorch implementation