search

forresti / SqueezeNet

SqueezeNet: AlexNet-level accuracy with 50x fewer parameters
BSD 2-Clause "Simplified" License
2.17k stars 723 forks source link

SqueezeNet in PyTorch #34

Closed Maratyszcza closed 7 years ago

Maratyszcza commented 7 years ago

SqueezeNet 1.0 and 1.1 are now available as a built-in model in the official pytorch/vision repo. In addition to model implementation, I pre-trained SqueezeNet 1.0 and 1.1 on ImageNet for PyTorch model zoo, and accuracy is even slightly better than the original Caffe models:

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

Links: code in the repo, discussion is in the PR

forresti commented 7 years ago

Bravo!!!

Maratyszcza commented 7 years ago

@forresti Would you add a link to readme?

forresti commented 7 years ago

Done!

Maratyszcza commented 7 years ago

Thanks!