Model Zoo¶
本页列出了经过预训练和预打包的模型存档，可以使用 TorchServe 进行推理。 要提出要包含的模型，请提交拉取请求。
特别感谢 PyTorch 社区，他们的 Model Zoo 和 Model Examples 用于生成这些模型存档。
| 型 | 类型 | 数据 | 大小 | 下载 | 样本输入 | 模型模式 |
|---|---|---|---|---|---|---|
| 亚历克斯网络 | 图像分类 | 图像网 | 216 兆字节 | 。三月 | kitten.jpg | 渴望 |
| 密集网161 | 图像分类 | 图像网 | 106 兆字节 | 。三月 | kitten.jpg | 渴望 |
| 雷斯网18 | 图像分类 | 图像网 | 41 兆字节 | 。三月 | kitten.jpg | 渴望 |
| VGG16的 | 图像分类 | 图像网 | 489 兆字节 | 。三月 | kitten.jpg | 渴望 |
| 挤压网 1_1 | 图像分类 | 图像网 | 4.4 兆字节 | 。三月 | kitten.jpg | 渴望 |
| MNIST 数字分类器 | 图像分类 | MNIST | 4.3 兆字节 | 。三月 | 0.png | 渴望 |
| 雷斯内特 152 | 图像分类 | 图像网 | 214 兆字节 | 。三月 | kitten.jpg | 渴望 |
| 更快的 RCNN | 对象检测 | 可可 | 148 兆字节 | 。三月 | persons.jpg | 渴望 |
| 面罩 RCNN | 对象检测 | 可可 | 158 兆字节 | 。三月 | persons.jpg | 渴望 |
| 文本分类器 | 文本分类 | AG_NEWS | 169 兆字节 | 。三月 | sample_text.txt | 渴望 |
| FCN ResNet 101 | 图像分割 | 可可 | 193 兆字节 | 。三月 | persons.jpg | 渴望 |
| DeepLabV3 ResNet 101 技术 | 图像分割 | 可可 | 217 兆字节 | 。三月 | persons.jpg | 渴望 |
| BERT 令牌分类 | 令牌分类 | AG_NEWS | 384.7 兆字节 | 。三月 | sample_text.txt | 渴望 |
| BERT 序列分类 | 序列分类 | AG_NEWS | 386.8 兆字节 | 。三月 | sample_text.txt | 渴望 |
| AlexNet 脚本 | 图像分类 | 图像网 | 216 兆字节 | 。三月 | kitten.jpg | Torch脚本 |
| Densenet161 脚本 | 图像分类 | 图像网 | 105 兆字节 | 。三月 | kitten.jpg | Torch脚本 |
| Resnet18 脚本化 | 图像分类 | 图像网 | 42 兆字节 | 。三月 | kitten.jpg | Torch脚本 |
| VGG16 脚本 | 图像分类 | 图像网 | 489 兆字节 | 。三月 | kitten.jpg | Torch脚本 |
| Squeezenet 1_1 脚本 | 图像分类 | 图像网 | 4.4 兆字节 | 。三月 | kitten.jpg | Torch脚本 |
| MNIST 数字分类器 脚本化 | 图像分类 | MNIST | 4.3 兆字节 | 。三月 | 0.png | Torch脚本 |
| Resnet 152 脚本 | 图像分类 | 图像网 | 215 兆字节 | 。三月 | kitten.jpg | Torch脚本 |
| 文本分类器 脚本化 | 文本分类 | AG_NEWS | 169 兆字节 | 。三月 | sample_text.txt | Torch脚本 |
| FCN ResNet 101 脚本 | 图像分割 | 可可 | 193 兆字节 | 。三月 | persons.jpg | Torch脚本 |
| DeepLabV3 ResNet 101 脚本 | 图像分割 | 可可 | 217 兆字节 | 。三月 | persons.jpg | Torch脚本 |
| MMF 活动识别 | 活动识别 | 哑谜 | 549 兆字节 | 。三月 | 372CC.mp4 | Torch脚本 |
| BERT 序列分类 CPU | 序列分类 | AG_NEWS | 386.9 兆字节 | 。三月 | sample_text.txt | Torch脚本 |
| BERT 序列分类 mGPU | 序列分类 | CAPTUM （盖图姆） | 386.9 兆字节 | 。三月 | sample_text_captum_input.txt | Torch脚本 |
| BERT 序列分类 | 序列分类 | AG_NEWS | 386.8 兆字节 | 。三月 | sample_text.txt | Torch脚本 |
| 犬种分类 | 图像分类 | 图像网 | 1.1 KB | 。战争 | kitten_small.jpg | 工作流 |
有关上述型号的更多详细信息，请参阅示例。