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

Add pre-trained models from HuggingFace and add accuracy testing #164

Merged
merged 9 commits into from
Jun 19, 2022

Conversation

darsnack
Copy link
Member

@darsnack darsnack commented Jun 9, 2022

This PR builds on the great work by @Alexander-Barth in FluxML/MetalheadWeights#2 to add pre-trained weights for VGG and ResNet from torchvision.

We ran into issues with Github LFS limits on MetalheadWeights. Instead, we are using HuggingFace to host our models now which is really ideal for our use case.

This PR adds:

  • Pre-trained models for VGG (w/o BN) + ResNet
  • Updated quickstart guide
  • Updated contributor docs for how to others can add weights via HF
  • Refactored tests to check the accuracy of pre-trained models (by just checking against a single reference image from ImageNet)

Close #86, closes #72, and closes #143.

test/runtests.jl Outdated Show resolved Hide resolved
@darsnack darsnack merged commit 81cef86 into FluxML:master Jun 19, 2022
@darsnack darsnack deleted the pretrain branch June 19, 2022 16:31
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
2 participants