Skip to content

SqueezeNet 1.0 and 1.1 models#49

Merged
apaszke merged 6 commits intopytorch:masterfrom
Maratyszcza:squeezenet
Feb 11, 2017
Merged

SqueezeNet 1.0 and 1.1 models#49
apaszke merged 6 commits intopytorch:masterfrom
Maratyszcza:squeezenet

Conversation

@Maratyszcza
Copy link
Copy Markdown
Contributor

@Maratyszcza Maratyszcza commented Jan 28, 2017 •
edited
Loading

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
Copy Markdown
Contributor Author

Maratyszcza commented Feb 3, 2017

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
Copy Markdown
Contributor

apaszke commented Feb 3, 2017

Awesome! Is the PR ready?

@Maratyszcza
Copy link
Copy Markdown
Contributor Author

Maratyszcza commented Feb 3, 2017 •
edited
Loading

There is SqueezeNet 1.1 to train as well

@Maratyszcza
Copy link
Copy Markdown
Contributor Author

Maratyszcza commented Feb 5, 2017

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
Copy Markdown
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
Copy Markdown
Contributor

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
Copy Markdown
Contributor Author

Maratyszcza commented Feb 7, 2017

@apaszke @soumith Ready to merge!

apaszke
apaszke reviewed Feb 8, 2017
View reviewed changes
Copy link
Copy Markdown
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.

Comment thread torchvision/models/squeezenet.py Outdated
)
self.classifier = nn.Sequential(
nn.Dropout(p=0.5),
self.final_conv,

This comment was marked as off-topic.

Sign in to view

This comment was marked as off-topic.

Sign in to view

This comment was marked as off-topic.

Sign in to view

Comment thread torchvision/models/squeezenet.py Outdated
nn.AvgPool2d(13)
)

import math

This comment was marked as off-topic.

Sign in to view

This comment was marked as off-topic.

Sign in to view

@apaszke apaszke merged commit d44273b into pytorch:master Feb 11, 2017
This was referenced Feb 11, 2017
Closed
Merged
@fmassa fmassa mentioned this pull request Sep 17, 2017
Open
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

1 more reviewer

@apaszke apaszke apaszke approved these changes

Reviewers whose approvals may not affect merge requirements

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

3 participants

@Maratyszcza @apaszke @soumith