-
Notifications
You must be signed in to change notification settings - Fork 27.7k
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
Feature to convert videomae huge and small finetuned on kinetics and ssv2 added to the videomae to pytorch converter #22788
Feature to convert videomae huge and small finetuned on kinetics and ssv2 added to the videomae to pytorch converter #22788
Conversation
…l finetuned kinetics and ssv2 added to videomae to pytorch converter
The documentation is not available anymore as the PR was closed or merged. |
c3ed1de
to
84647ee
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for adding! Would it be possible to upload the huge checkpoints to the hub under your personal user account? Then I'll transfer them to the appropriate organization.
@NielsRogge Uploaded the huge-kinetics, small-kinetics, and small-ssv2 models to the HF model hub under the following names:
Let me know if anything else is needed. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for your contribution!
…ssv2 added to the videomae to pytorch converter (huggingface#22788) * Feature to convert videomae huge finetuned kinetics and videomae small finetuned kinetics and ssv2 added to videomae to pytorch converter * Reformat convert_videomae_to_pytorch using black * Value exception added for the possible videomae model architectures
What does this PR do?
Adds the feature to convert VideoMAE huge and small to hugging face compatible pytorch model. The following models are added to the converter:
Who can review?
@NielsRogge