GithubHelp home page GithubHelp logo

transmuteai / chipnet Goto Github PK

View Code? Open in Web Editor NEW
20.0 4.0 7.0 756 KB

Official implementation "ChipNet: Budget-Aware Pruning with Heaviside Continuous Approximations"

License: MIT License

Python 98.83% Shell 1.17%
iclr2021 pruning structured continuous-relexation heaviside budget-aware

chipnet's People

Contributors

arnav0400 avatar rishabh-16 avatar ubamba98 avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar

chipnet's Issues

Extract pruned model for plain inference

Hi, amazing work guys. I was wondering if you have some code readily available to extract the fine-tuned pruned sub-model out of the checkpoint model, so that it can be used by simple inference scripts?

Implementation of Network Slimming

Hi,

Is it possible to release the code for your network slimming implementation which is reported in the paper. The original code seems to have problems with large pruning ratio.

Thanks.

Pruning with Uneven Block Dimensions

Hi,

We just came across your work, it was amazing!

We have a question about the ResNet pruning that when we run the ChipNet code and get these results.

Test Acc: 0.7316
Total Trainable Params: 317198
Channels pruned: 80.00%
Parameters pruned: 81.63%
image

Test Acc: 0.7556
Total Trainable Params: 853033
Channels pruned: 60.00%
Parameters pruned: 50.59%
image

We saw that the output channels for the blocks are not the same, so how can we get a smaller model with it?
Did you do padding to make sure they are the same?

Cannot reproduce results in the paper

Please refer to the CIFAR-10, Resnet-50 architecture in the table 2 of paper.

When volume budget is 12.5 %, number of parameter is 2.8% and FLOPs is 5.1% in Table 2.

However, the my reproduced result after pruning step is the number of parameter is 12.9% and FLOPs is 16.9 %
while the volume budget is 12.5%.

I follow the hyperparameter setting provided in the appendix.

Would you share train-set validation-set split information and pretrained networks to reproduce the results in the paper??

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.