Skip to content

Latest commit

 

History

History
32 lines (27 loc) · 5.41 KB

model_zoo.md

File metadata and controls

32 lines (27 loc) · 5.41 KB

Model Zoo

This page lists model archives that are pre-trained and pre-packaged, ready to be served for inference with TorchServe. To propose a model for inclusion, please submit a pull request.

Special thanks to the PyTorch community whose Model Zoo and Model Examples were used in generating these model archives.

Model Type Dataset Size Download Sample Input Model mode
AlexNet Image Classification ImageNet 216 MB .mar kitten.jpg Eager
Densenet161 Image Classification ImageNet 106 MB .mar kitten.jpg Eager
Resnet18 Image Classification ImageNet 41 MB .mar kitten.jpg Eager
VGG11 Image Classification ImageNet 471 MB .mar kitten.jpg Eager
Squeezenet 1_1 Image Classification ImageNet 4.4 MB .mar kitten.jpg Eager
MNIST digit classifier Image Classification MNIST 4.3 MB .mar 0.png Eager
Resnet 152 Image Classification ImageNet 214 MB .mar kitten.jpg Eager
Faster RCNN Object Detection COCO 148 MB .mar persons.jpg Eager
MASK RCNN Object Detection COCO 158 MB .mar persons.jpg Eager
Text classifier Text Classification AG_NEWS 169 MB .mar sample_text.txt Eager
FCN Resenet 101 Image Segmentation COCO 193 MB .mar persons.jpg Eager
AlexNet Scripted Image Classification ImageNet 216 MB .mar kitten.jpg Torchscripted
Densenet161 Scripted Image Classification ImageNet 105 MB .mar kitten.jpg Torchscripted
Resnet18 Scripted Image Classification ImageNet 42 MB .mar kitten.jpg Torchscripted
VGG11 Scripted Image Classification ImageNet 471 MB .mar kitten.jpg Torchscripted
Squeezenet 1_1 Scripted Image Classification ImageNet 4.4 MB .mar kitten.jpg Torchscripted
MNIST digit classifier Scripted Image Classification MNIST 4.3 MB .mar 0.png Torchscripted
Resnet 152 Scripted Image Classification ImageNet 215 MB .mar kitten.jpg Torchscripted
Text classifier Scripted Text Classification AG_NEWS 169 MB .mar sample_text.txt Torchscripted
FCN Resenet 101 Scripted Image Segmentation COCO 193 MB .mar persons.jpg Torchscripted

Refer example for more details on above models.