Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Channel-based pruning methods #26

Open
annahambi opened this issue Apr 1, 2022 · 0 comments
Open

Channel-based pruning methods #26

annahambi opened this issue Apr 1, 2022 · 0 comments

Comments

@annahambi
Copy link

Dear shrinkbench authors

The channel based pruning strategies seem unsupported. Even though the code is there, it is commented out in the strategies.utils.py and not imported in the strategies/__init__.py. Can you comment on the reason or if there is a plan to activate this strategies?

for strategy in ['WeightNormChannelPruning', 
                'ActivationNormChannelPruning']:

    for  c in [1,2,4,8,16,32,64]:

        exp = PruningExperiment(dataset='MyCIFAR10', 
                                model='my_vgg16',
                                strategy=strategy,
                                compression=c,
                                train_kwargs={'epochs':10}, 
                                save_freq=9, 
                                pretrained=False)
        exp.run()
        clear_output()

==> AttributeError: module 'shrinkbench.strategies' has no attribute 'WeightNormChannelPruning'```
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant