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.

Sign up for GitHub

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

Jump to bottom

Question about vitprune.py code #3

Open

wonkicho opened this issue Sep 23, 2022 · 1 comment

wonkicho commented Sep 23, 2022

hello

i wonder what does mean about '[16,40,64,88,112,136]' values of list!!

haodonglu commented Nov 29, 2022

hello

i wonder what does mean about '[16,40,64,88,112,136]' values of list!!

These values denote the layer index for Attention, and the pruning ratio needs to be adjusted to ensure that dim is divisible by heads.

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