Add pretrained weights for MobileNet v2 and v3#288
Open
paulnovo wants to merge 3 commits intoFluxML:masterfrom
Open
Add pretrained weights for MobileNet v2 and v3#288paulnovo wants to merge 3 commits intoFluxML:masterfrom
paulnovo wants to merge 3 commits intoFluxML:masterfrom
Conversation
Member
|
I create an huggingface repo where you can upload the weights |
Author
Thank you @CarloLucibello, I have uploaded the weights and updated the PR. I think this is good to go, but let me know if anything else is needed. |
* Set epsilon to 0.001 and momentum to 0.01 in all BatchNorm layers * MobileNet V3 uses a ReLu activation in all squeeze excitation layers * Fix stride in final mbconv block of both small and large models, this should be set to 2 instead of 1
* Adds pre-trained weights for MobileNet v3 small and large models * Adds pre-trained weights for MobileNet v2 * Adds, commented, lines to the model_list in the port_torchvision.jl script for generating these weights from torchvision models * Replaces the test image (of a guitar) in the pytorch2flux.jl script, the link is currently unavailable
* Add models to PRETRAINED_MODELS to enable acctest * Update pretrained status in README
537b258 to
a5e080d
Compare
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
This PR adds pretrained weights for MobileNet v2 and MobileNet v3 (both small and large) using the
port_torchvision.jlscript to generate the weights from torchvision. The PR also fixes MobileNet v3 in Metalhead to match the original paper and torchvision implementation.Adding pretrained weights checklist