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 File | Type | Dataset | Size | Download |
|---|---|---|---|---|
| AlexNet | Image Classification | ImageNet | 216 MB | .mar |
| Densenet161 | Image Classification | ImageNet | 106 MB | .mar |
| Resnet18 | Image Classification | ImageNet | 41 MB | .mar |
| VGG11 | Image Classification | ImageNet | 471 MB | .mar |
| Squeezenet 1_1 | Image Classification | ImageNet | 4.4 MB | .mar |
| MNIST digit classifier | Image Classification | MNIST | 4.3 MB | .mar |
| Resnet 152 | Image Classification | ImageNet | 214 MB | .mar |
| Faster RCNN | Object Detection | COCO | 148 MB | .mar |
| MASK RCNN | Object Detection | COCO | 158 MB | .mar |
| Text classifier | Text Classification | AG_NEWS | 169 MB | .mar |
| FCN Resenet 101 | Image Segmentation | COCO | 193 MB | .mar |
| Refer example for more details on above models. |