Skip to content

Does this script run with the hybrid-filled method by default?Thanks! #41

Open
@iMountTai

Description

@iMountTai

Activity

sangamesh-kodge

sangamesh-kodge commented on Jun 5, 2023

@sangamesh-kodge

I had the same question. I understand I need to set "attention_pruning_method" : pruning_method and "dense_pruning_method" : pruning_method:1d_alt for hybrid pruning.

Is it that for hybrid-filled approach we train the model again with reinitializing the weights for heads which are not completely pruned and this is done separately after hybrid pruning? or Is there a argument I can set in the code for hybrid-filled pruning? @madlag @echarlaix

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

      Development

      No branches or pull requests

        Participants

        @iMountTai@sangamesh-kodge

        Issue actions

          Does this script run with the hybrid-filled method by default?Thanks! · Issue #41 · huggingface/nn_pruning