模型库¶
此页面列出了经过预训练和预打包的模型存档，已准备好用于 TorchServe 的推理服务。要提议将模型包含在内，请提交拉取请求。
特别感谢 PyTorch 社区，其模型库和模型示例被用于生成这些模型存档。
| 模型 | 类型 | 数据集 | 大小 | 下载 | 样本输入 | 模型模式 |
|---|---|---|---|---|---|---|
| AlexNet | 图像分类 | ImageNet | 216 MB | .mar | kitten.jpg | Eager |
| Densenet161 | 图像分类 | ImageNet | 106 MB | .mar | kitten.jpg | Eager |
| Resnet18 | 图像分类 | ImageNet | 41 MB | .mar | kitten.jpg | Eager |
| VGG16 | 图像分类 | ImageNet | 489 MB | .mar | kitten.jpg | Eager |
| Squeezenet 1_1 | 图像分类 | ImageNet | 4.4 MB | .mar | kitten.jpg | Eager |
| MNIST 数字分类器 | 图像分类 | MNIST | 4.3 MB | .mar | 0.png | Eager |
| Resnet 152 | 图像分类 | ImageNet | 214 MB | .mar | kitten.jpg | Eager |
| Faster RCNN | 目标检测 | COCO | 148 MB | .mar | persons.jpg | Eager |
| MASK RCNN | 目标检测 | COCO | 158 MB | .mar | persons.jpg | Eager |
| 文本分类器 | 文本分类 | AG_NEWS | 169 MB | .mar | sample_text.txt | Eager |
| FCN ResNet 101 | 图像分割 | COCO | 193 MB | .mar | persons.jpg | Eager |
| DeepLabV3 ResNet 101 | 图像分割 | COCO | 217 MB | .mar | persons.jpg | Eager |
| BERT 令牌分类 | 令牌分类 | AG_NEWS | 384.7 MB | .mar | sample_text.txt | Eager |
| BERT 序列分类 | 序列分类 | AG_NEWS | 386.8 MB | .mar | sample_text.txt | Eager |
| AlexNet 脚本化 | 图像分类 | ImageNet | 216 MB | .mar | kitten.jpg | Torchscripted |
| Densenet161 脚本化 | 图像分类 | ImageNet | 105 MB | .mar | kitten.jpg | Torchscripted |
| Resnet18 脚本化 | 图像分类 | ImageNet | 42 MB | .mar | kitten.jpg | Torchscripted |
| VGG16 脚本化 | 图像分类 | ImageNet | 489 MB | .mar | kitten.jpg | Torchscripted |
| Squeezenet 1_1 脚本化 | 图像分类 | ImageNet | 4.4 MB | .mar | kitten.jpg | Torchscripted |
| MNIST 数字分类器脚本化 | 图像分类 | MNIST | 4.3 MB | .mar | 0.png | Torchscripted |
| Resnet 152 脚本化 | 图像分类 | ImageNet | 215 MB | .mar | kitten.jpg | Torchscripted |
| 文本分类器脚本化 | 文本分类 | AG_NEWS | 169 MB | .mar | sample_text.txt | Torchscripted |
| FCN ResNet 101 脚本化 | 图像分割 | COCO | 193 MB | .mar | persons.jpg | Torchscripted |
| DeepLabV3 ResNet 101 脚本化 | 图像分割 | COCO | 217 MB | .mar | persons.jpg | Torchscripted |
| MMF 活动识别 | 活动识别 | Charades | 549 MB | .mar | 372CC.mp4 | Torchscripted |
| BERT 序列分类 CPU | 序列分类 | AG_NEWS | 386.9 MB | .mar | sample_text.txt | Torchscripted |
| BERT 序列分类 mGPU | 序列分类 | CAPTUM | 386.9 MB | .mar | sample_text_captum_input.txt | Torchscripted |
| BERT 序列分类 | 序列分类 | AG_NEWS | 386.8 MB | .mar | sample_text.txt | Torchscripted |
| 狗品种分类 | 图像分类 | ImageNet | 1.1 KB | .war | kitten_small.jpg | 工作流 |
有关上述模型的更多详细信息，请参考示例。
