Skip to content

SqueezeNet 1.0 and 1.1 models #49

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

Merged
merged 6 commits into from
Feb 11, 2017
Merged

SqueezeNet 1.0 and 1.1 models #49

merged 6 commits into from
Feb 11, 2017

Conversation

Maratyszcza
Copy link
Contributor

@Maratyszcza Maratyszcza commented Jan 28, 2017

SqueezeNet 1.0 and 1.1 models. I'm still working on getting them converge on ImageNet, but as several people contacted me about SqueezeNet for PyTorch, I think its useful to share it publicly.

@Maratyszcza
Copy link
Contributor Author

I successfully trained SqueezeNet 1.0 on ImageNet with the following parameters: --arch squeezenet1_0 --epochs 90 --lr 0.01 --weight-decay 0.0002 --batch-size 128.
Accuracy: Prec@1 57.868 Prec@5 80.424

@apaszke
Copy link
Contributor

apaszke commented Feb 3, 2017

Awesome! Is the PR ready?

@Maratyszcza
Copy link
Contributor Author

Maratyszcza commented Feb 3, 2017

There is SqueezeNet 1.1 to train as well

@Maratyszcza
Copy link
Contributor Author

Finished training both SqueezeNet 1.0 and 1.1. Trained for 100 epochs with --epochs 100 --lr 0.01 --weight-decay 0.0002 --batch-size 128

Model Top-1 accuracy Top-5 accuracy
SqueezeNet 1.0 58.000% 80.488%
SqueezeNet 1.1 58.184% 80.514%

After @soumith uploads the models to AWS, I'll add model_zoo links, and we'd be ready to merge.

@apaszke
Copy link
Contributor

apaszke commented Feb 6, 2017

Great! If you put them somewhere or send them to me, I can upload them to AWS too.

@soumith
Copy link
Member

soumith commented Feb 6, 2017

finished uploading them:

https://s3.amazonaws.com/pytorch/models/squeezenet1_0-0ef9bd12.pth.tar
https://s3.amazonaws.com/pytorch/models/squeezenet1_1-f0e1d24d.pth.tar

@Maratyszcza Maratyszcza changed the title [WIP] SqueezeNet 1.0 and 1.1 models SqueezeNet 1.0 and 1.1 models Feb 6, 2017
@Maratyszcza
Copy link
Contributor Author

@apaszke @soumith Ready to merge!

Copy link
Contributor

@apaszke apaszke left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Two minor things. Ready to merge after fixing them.

)
self.classifier = nn.Sequential(
nn.Dropout(p=0.5),
self.final_conv,

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

nn.AvgPool2d(13)
)

import math

This comment was marked as off-topic.

This comment was marked as off-topic.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants