-
Notifications
You must be signed in to change notification settings - Fork 81
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
Could you share the pretrained model for imagenet? #17
Comments
Hi, We used pre-trained models, which can be downloaded from https://pytorch.org/docs/stable/torchvision/models.html following our code. However, we haven't published the SWAG checkpoints at the end of training due to their size --- @timgaripov might be able to provide them if he still has them. |
@wjmaddox could you release the training scripts for imagenet? like lr config, .. |
using python experiments/imagenet/run_swag_imagenet.py |
That sounds about right given the hyper-parameters in Appendix H of https://arxiv.org/pdf/1902.02476.pdf |
I hope to research your proposal, SWGA for imagenet.
Could you share the pretrained model for imagenet?
The text was updated successfully, but these errors were encountered: