---

VastModelZOO是瀚博半导体维护的AI模型平台，提供了人工智能多个领域（CV、AUDIO、NLP、LLM、MLLM等）的开源模型在瀚博训推芯片上的部署、训练示例。

VastModelZOO旨在基于瀚博半导体的硬件产品和软件SDK，展示最佳的编程实践，以达成模型的快速移植和最优性能。

为方便大家使用VastModelZOO，我们将持续增加典型模型和基础插件。

依赖软件

• 瀚博推理引擎类型：

  • Build_In: 瀚博自研软件栈推理引擎
    • 可通过拉取docker快速在瀚博硬件产品上进行验证测试
  • PyTorch: VACC Extension for PyTorch插件

• 版本说明

  组件	版本	工具说明
  Driver	3.3.0	PCIe 驱动包
  AI-Release	AI3.0_SP9_0811	AI 工具包
  VAMC	3.4.1	模型转换工具
  VastPipe	2.7.3	全流程低代码开发框架
  VastStreamX	2.8.3	SDK应用API库

  Driver及工具文档下载链接: Baidu Netdisk

模型列表

CV Models

• classification

model	codebase	model list	model type	runtime
CSPNet	timm	model name cspresnet50 cspresnext50 cspdarknet53	classification	Build_In
CSPNet	mmcls	model name cspresnet50 cspresnext50 cspdarknet53	classification	Build_In
CSPNet	ppcls	model name cspdarknet53	classification	Build_In
EfficientNet	timm	model name efficientnet_b0 efficientnet_b1 efficientnet_b2 efficientnet_b3 efficientnet_b4 tf_efficientnet_b5 tf_efficientnet_b6 tf_efficientnet_b7 tf_efficientnet_b8	classification	Build_In
EfficientNet_v2	timm	model name efficientnetv2_rw_t efficientnetv2_rw_s efficientnetv2_rw_m	classification	Build_In
MobileNet_v2	torchvision	model name mobilenetv2	classification	Build_In
MobileNet_v2	timm	model name mobilenetv2	classification	Build_In
MobileNet_v2	mmcls	model name mobilenetv2	classification	Build_In
MobileNet_v2	keras	model name mobilenetv2	classification	Build_In
MobileNet_v2	ppcls	model name mobilenet_v2_x0.25 mobilenet_v2_x0.5 mobilenet_v2_x0.75 mobilenet_v2 mobilenet_v2_x1.5 mobilenet_v2_x2.0 mobilenet_v2_ssld	classification	Build_In
MobileNet_v3	torchvision	model name mobilenet_v3_small mobilenet_v3_large	classification	Build_In
MobileNet_v3	timm	model name mobilenet_v3_x1.0 mobilenet_v3_small_x0.5 mobilenet_v3_small_x0.75 mobilenet_v3_small_x1.0 mobilenet_v3_large_x1.0	classification	Build_In
MobileNet_v3	ppcls	model name mobilenet_v3_small_x0.35 mobilenet_v3_small_x0.35_ssld mobilenet_v3_small_x0.5 mobilenet_v3_small_x0.75 mobilenet_v3_small_x1.0 mobilenet_v3_small_x1.0_ssld mobilenet_v3_small_x1.25 mobilenet_v3_large_x0.35 mobilenet_v3_large_x0.5 mobilenet_v3_large_x0.75 mobilenet_v3_large_x1.0 mobilenet_v3_large_x1.0_ssld mobilenet_v3_large_x1.25	classification	Build_In
MobileNet_v3	showlo	model name mobilenet_v3_small	classification	Build_In
MobileNet_v3	sqlai	model name mobilenet_v3_small mobilenet_v3_large	classification	Build_In
MobileViT	apple	model name mobilevit-s	classification	Build_In
RepOPT	official	model name RepOpt-VGG-B1 RepOpt-VGG-B2 RepOpt-VGG-L1 RepOpt-VGG-L2	classification	Build_In
RepVGG	official	model name RepVGG-A0 RepVGG-A1 RepVGG-A2 RepVGG-B0 RepVGG-B1 RepVGG-B2 RepVGG-B1g2 RepVGG-B1g4 RepVGG-B2g4 RepVGG-B3 RepVGG-B3g4	classification	Build_In
ResNeSt	official	model name resnest50 resnest101 resnest200 resnest269	classification	Build_In
ResNet	timm	model name resnet18 resnet26 resnet34 resnet50 resnet101 resnet152 gluon_resnet18_v1b gluon_resnet34_v1b gluon_resnet50_v1b gluon_resnet50_v1c gluon_resnet50_v1d gluon_resnet50_v1s gluon_resnet101_v1b gluon_resnet101_v1c gluon_resnet101_v1d gluon_resnet101_v1s gluon_resnet152_v1b gluon_resnet152_v1c gluon_resnet152_v1d gluon_resnet152_v1s	classification	Build_In
ResNet	torchvision	model name resnet18 resnet34 resnet50 resnet101 resnet152	classification	Build_In
ResNet	mmcls	model name resnet18 resnet34 resnet50 resnet101 resnet152	classification	Build_In
ResNet	ppcls	model name resnet18 resnet18_vd resnet34 resnet34_vd resnet34_vd_ssld resnet50 resnet50_vc resnet50_vd resnet50_vd_ssld resnet101 resnet101_vd resnet101_vd_ssld resnet152 resnet152_vd resnet200_vd	classification	Build_In
ResNet	keras	model name resnet50 resnet50v2 resnet101 resnet101v2 resnet152 resnet152v2	classification	Build_In
ResNet	oneflow	model name resnet18 resnet34 resnet50 resnet101 resnet152	classification	Build_In
ResNeXt	timm	model name resnext50_32x4d resnext50d_32x4d resnext101_32x8d resnext101_64x4d tv_resnext50_32x4d	classification	Build_In
ResNeXt	torchvision	model name resnext50_32x4d resnext101_32x8d	classification	Build_In
ResNeXt	mmcls	model name resnext50_32x4d resnext101_32x4d resnext101_32x8d resnext152_32x4d	classification	Build_In
ResNeXt	ppcls	model name resnext50_32x4d resnext50_64x4d resnext50_vd_32x4d resnext50_vd_64x4d resnext101_32x4d resnext101_64x4d resnext101_vd_32x4d resnext101_vd_64x4d resnext152_32x4d resnext152_64x4d resnext152_vd_32x4d resnext152_vd_64x4d resnext101_32x8d_wsl resnext101_32x16d_wsl resnext101_32x32d_wsl	classification	Build_In
SENet	timm	model name seresnet50 seresnet152d seresnext26d_32x4d seresnext26t_32x4d seresnext50_32x4d seresnext101_32x8d seresnext101d_32x8d legacy_seresnet18 legacy_seresnet34 legacy_seresnet50 legacy_senet154 legacy_seresnext26_32x4d legacy_seresnext50_32x4d legacy_seresnext101_32x4d legacy_seresnet101 legacy_seresnet152	classification	Build_In
Swin	microsoft	model name swin-b	classification	Build_In
ViT	huggingface	model name vit-base-patch16-224	classification	Build_In

• object detection

model	codebase	model list	model type	runtime
CenterNet	official	model name centernet_res18	object detection	Build_In
DETR	facebook	model name detr_res50	object detection	Build_In
Grounding-DINO	official	model name groundingdino_swint_ogc groundingdino_swinb_cogcoor	object detection	Build_In
RT-DETR	official	model name rtdetr_r18vd	object detection	Build_In
YOLO-World	official	model name yolo_world_v2_l	object detection	Build_In
Yolov3	pytorch(u)	model name yolov3 yolov3-spp yolov3-tiny	object detection	Buid_In
Yolov4	darknet	model name yolov4 yolov4_tiny yolov4_csp yolov4_csp_swish yolov4_csp_x_swish yolov4x_mish	object detection	Buid_In
Yolov4	bubbliiiing	model name yolov4 yolov4_tiny	object detection	Buid_In
Yolov4	tianxiaomo	model name yolov4	object detection	Buid_In
Yolov5	pytorch(u)	model name yolov5n yolov5s yolov5m yolov5l yolov5x yolov5n6 yolov5s6 yolov5m6 yolov5l6 yolov5x6	object detection	Build_In
Yolov5	mmyolo	model name yolov5n yolov5s yolov5m yolov5l yolov5n6 yolov5s6 yolov5m6 yolov5l6	object detection	Build_In
yolov6	yolov6	model name YOLOv6-n YOLOv6-tiny YOLOv6-s	object detection	Build_In
yolov7	yolov7	model name YOLOv7 YOLOv7x YOLOv7-w6 YOLOv7-e6 YOLOv7-d6 YOLOv7-e6e	object detection	Build_In
yolov8	yolov8	model name YOLOv8n YOLOv8s YOLOv8m YOLOv8b YOLOv8l YOLOv8x	object detection	Build_In
yolov10	yolov10	model name YOLOv10-N YOLOv10-S YOLOv10-M YOLOv10-B YOLOv10-L YOLOv10-X	object detection	Build_In
Yolov11	official	model name yolo11n yolo11s yolo11m yolo11l yolo11x	object detection	Build_In
Yolov12	official	model name yolov12n yolov12s yolov12m yolov12l yolov12x	object detection	Build_In
Yolox	official	model name yolox_s yolox_m yolox_l yolox_x yolox_darknet yolox_tiny yolox_nano	object detection	Build_In
Yolox	mmyolo	model name yolox_s yolox_tiny	object detection	Build_In

• segmentation

model	codebase	model list	model type	runtime
BiSeNet	pytorch	model name bisenetv1 bisenetv2	segmentation	Build_In
BiSeNet	pytorch	model name bisenet bisenet_2class	face segmentation	Build_In
Deeplab_v3	pytorch	model name deeplabv3_resnet50 deeplabv3_resnet101	segmentation	Build_In
Deeplab_v3_plus	pytorch	model name deeplabv3plus_resnet50 deeplabv3plus_resnet101	segmentation	Build_In
FCN	pytorch	model name fcn8s fcn16s fcn32s	segmentation	Build_In
FCN	mmseg	model name fcn_r50_d8_20k fcn_r50_d8_40k fcn_r101_d8_20k fcn_r101_d8_40k	segmentation	Build_In
Human_Seg	pytorch	model name unet_resnet18 deeplabv3plus_resnet18	human segmentation	Build_In
Mask2Former	official	model name maskformer2_R50	instance segmentation	Build_In
MODNet	official	model name modnet	matting	Build_In
Unet	bubbliiiing	model name unet_vgg16 unet_resnet50	segmentation	Build_In
Unet	milesial	model name unet_scale0.5 unet_scale1.0	segmentation	Build_In
Unet	keras	model name unet	segmentation	Build_In
Unet3P	pytorch	model name unet3p unet3p_deepsupervision	segmentation	Build_In
UnetPP	pytorch	model name unetpp	segmentation	Build_In
Yolov8-seg	ultralytics	model name yolov8n-seg yolov8s-seg yolov8m-seg yolov8l-seg yolov8x-seg	instance segmentation	Build_In

• face alignment

model	codebase	model list	model type	runtime
PIPNet	official	model name pip_resnet18 pip_resnet101 pipnet_mobilenet_v2 pipnet_mobilenet_v3	face alignment	Build_In
PFLD	pytorch	model name pfld	face alignment	Build_In

• face detection

model	codebase	model list	model type	runtime
RetinaFace	pytorch	model name retinaface-resnet50 retinaface-mobilenet0.25	face detection	Build_In
SCRFD	insightface	model name scrfd_500m scrfd_500m_bnkps scrfd_1g scrfd_2.5g scrfd_2.5g_bnkps scrfd_10g scrfd_10g_bnkps scrfd_34g	face detection	Build_In

• facial quality

model	codebase	model list	model type	runtime
SDD-FIQA	sdd_fiqa	model name sdd_fiqa	face quality	Build_In

• face recognize

model	codebase	model list	model type	runtime
FaceNet	tensorflow	model name facenet_vggface2 facenet_casia_webface	face recognize	Build_In
FaceNet	pytorch	model name facenet_vggface2 facenet_casia_webface	face recognize	Build_In

• facial attribute

model	codebase	model list	model type	runtime
FairFace	official	model name fairface_res34	face attribute	Build_In

• image colorization

model	codebase	model list	model type	runtime
GPEN	official	model name gpen	image colorization	Build_In

• image retrieval

model	codebase	model list	model type	runtime
DINOv2	official	model name dinov2_vitl14_reg4	image_retrieval	Build_In

• low light image enhancement

model	codebase	model list	model type	runtime
PairLIE	official	model name pairlie	low light image enhancement	Build_In

• mot

model	codebase	model list	model type	runtime
ByteTrack	official	model name ByteTrack_ablation bytetrack_x_mot17 bytetrack_l_mot17 bytetrack_m_mot17 bytetrack_s_mot17 bytetrack_nano_mot17 bytetrack_tiny_mot17 bytetrack_x_mot20	mot	Build_In
DeepSort	pytorch	model name fast reid	mot	Build_In

• pose

model	codebase	model list	model type	runtime
OpenPose	pytorch	model name body model hand model	pose	Build_In
HRNet_Pose	mmpose	model name hrnet_pose	pose	Build_In
Yolov8_Pose	ultralytics	model name yolov8n_pose yolov8s_pose yolov8m_pose yolov8l_pose yolov8x_pose yolov8x_pose_p6	pose	Build_In

• reid

model	codebase	model list	model type	runtime
Fast_Reid	official	model name market_bot_R50 market_bot_S50 market_bot_R50_ibn market_bot_R101_ibn	reid	Build_In

• salient object detection

model	codebase	model list	model type	runtime
F3Net	official	model name f3net	salient object detection	Build_In
ISNet	official	model name isnet	salient object detection	Build_In

• super_resolution

model	codebase	model list	model type	runtime
EDSR	basicsr	model name edsr_1x edsr_m2x	super resolution	Build_In
EDSR	official	model name edsr_x2 edsr_baseline_x2 edsr_baseline_x4	super resolution	Build_In
GPEN	official	model name gpen	face super resolution	Build_In
NCNet	official	model name ncnet	super resolution	Build_In
RCAN	official	model name rcan rcan2	super resolution	Build_In
RCAN	basicsr	model name rcan	super resolution	Build_In
VDSR	pytorch	model name vdsr	super resolution	Build_In

• text detection

model	codebase	model list	model type	runtime
DBNet	ppocr	model name dbnet_mobilenet_v3 dbnet_resnet50_vd ch_PP_OCRv3_det ch_PP_OCRv4_det en_PP_OCRv3_det	text detection	Build_In
DBNet	official	model name dbnet_resnet18 dbnet_resnet50	text detection	Build_In
DBNet	mmocr	model name dbnet_resnet18_fpnc_1200e_icdar2015 dbnet_resnet18_fpnc_1200e_totaltext dbnet_resnet18_fpnc_100k_synthtext dbnet_resnet50_1200e_icdar2015 dbnet_resnet50_oclip_1200e_icdar2015	text detection	Build_In

• text recognition

model	codebase	model list	model type	runtime
CNN_CTC	pytorch	model name resnet_fc	ocr recognize	Build_In
CRNN	ppocr	model name resnet34_vd	text recognition	Build_In
PPOCR_V4_REC	ppocr	model name ppocr_v4_rec	ocr recognize	Build_In

NLP Models

• information extraction

model	codebase	model list	model type	runtime
uie	uie_pytorch	model name uie-base	Information extraction	Build_In

• named entity recognition

model	codebase	model list	model type	runtime
BERT	huggingface	model name bert_base_zh_ner-256	named entity recognition	Build_In
RoBERTa	huggingface	model name roberta_wwm_ext_base_zh-256	named entity recognition	Build_In

• question answering

model	codebase	model list	model type	runtime
BERT	tensorflow	model name bert_base_en_qa-384 bert_large_en_qa-384	sentence classification	Build_In
BERT	huggingface	model name bert_base_en_qa-384 bert_large_en_qa-384	sentence classification	Build_In

• sentence classification

model	codebase	model list	model type	runtime
BERT	tensorflow	model name bert_base_mrpc_cls-128 bert_large_mrpc_cls-128	sentence classification	Build_In
BERT	huggingface	model name bert_base_mrpc_cls-128 bert_base_mrpc_cls-512 bert_large_mrpc_cls-128 bert_large_mrpc_cls-512 bert_base_imdb_cls-128 bert_base_sst2_cls-128	sentence classification	Build_In
BERT	modelscope	model name nlp_structbert_sentiment-classification_chinese-base	sentence classification	Build_In
Electra	CIB	model name electra_small-512	sentence classification	Build_In
Electra	huggingface	model name electra_small_dc_mrpc-128 electra_small_gen_mrpc-128 electra_base_dc_mrpc-128 electra_base_gen_mrpc-128 electra_large_dc_mrpc-128	sentence classification	Build_In
RoBERTa	huggingface	model name roberta_base_en_cls-128	sentence classification	Build_In

• Text2Vec

model	codebase	model list	model type	runtime
bge	huggingface	model name bge-reranker-base bge-reranker-large bge-reranker-v2-m3	Reranker model	Build_In
bce	huggingface	model name bce-reranker-base_v1	Reranker model	Build_In

LLM Models

model	codebase	model list	model type	runtime
LLaMA	huggingface	model name meta-llama-33b	large language model	Build_In/PyTorch
LLaMA-2	huggingface	model name Llama-2-7b-hf Llama-2-7b-chat-hf Llama-2-13b-hf Llama-2-13b-chat-hf Llama-2-70b-hf Llama-2-70b-chat-hf	large language model	Build_In/PyTorch
LLaMA-3	huggingface	model name Meta-Llama-3-8B Meta-Llama-3-8B-Instruct Meta-Llama-3-70B Meta-Llama-3-70B-Instruct	large language model	Build_In/PyTorch
LLaMA-3.1	huggingface	model name Meta-Llama-3.1-8B Meta-Llama-3.1-8B-Instruct Meta-Llama-3.1-70B Meta-Llama-3.1-70B-Instruct	large language model	Build_In/PyTorch
LLaMA-3.2	huggingface	model name Llama-3.2-1B Llama-3.2-1B-Instruct Llama-3.2-3B Llama-3.2-3B-Instruct	large language model	Build_In/PyTorch
LLaMA-3.3	huggingface	model name Llama-3.3-70B-Instruct	large language model	Build_In/PyTorch
Internlm2	huggingface	model name internlm2-1_8b internlm2-chat-1_8b internlm2-7b internlm2-chat-7b internlm2-20b internlm2-chat-20b	large language model	Build_In/PyTorch
Internlm2_5	huggingface	model name internlm2_5-1_8b internlm2_5-1_8b-chat internlm2_5-7b internlm2_5-7b-chat internlm2_5-20b internlm2_5-20b-chat	large language model	Build_In/PyTorch
Internlm3	huggingface	model name internlm3-8b-instruct	large language model	Build_In/PyTorch
ChatGLM2	huggingface	model name chatglm2-6b	large language model	Build_In/PyTorch
ChatGLM3	huggingface	model name chatglm3-6b-base chatglm3-6b	large language model	Build_In/PyTorch
ChatGLM4	huggingface	model name glm-4-9b-hf glm-4-9b-chat-hf	large language model	Build_In/PyTorch
BaiChuan2	huggingface	model name Baichuan2-7B-Base Baichuan2-7B-Chat Baichuan2-13B-Base Baichuan2-13B-Chat	large language model	Build_In/PyTorch
Qwen1.5	huggingface	model name Qwen1.5-0.5B Qwen1.5-0.5B-Chat Qwen1.5-1.8B Qwen1.5-1.8B-Chat Qwen1.5-4B Qwen1.5-4B-Chat Qwen1.5-7B Qwen1.5-7B-Chat Qwen1.5-14B Qwen1.5-14B-Chat Qwen1.5-32B Qwen1.5-32B-Chat Qwen1.5-72B Qwen1.5-72B-Chat Qwen1.5-110B-Chat	large language model	Build_In/PyTorch
Qwen2	huggingface	model name Qwen2-0.5B Qwen2-0.5B-Instruct Qwen2-1.5B Qwen2-1.5B-Instruct Qwen2-7B Qwen2-7B-Instruct Qwen2-72B Qwen2-72B-Instruct	large language model	Build_In/PyTorch
Qwen2.5	huggingface	model name Qwen2.5-0.5B Qwen2.5-0.5B-Instruct Qwen2.5-1.5B Qwen2.5-1.5B-Instruct Qwen2.5-3B Qwen2.5-3B-Instruct Qwen2.5-7B Qwen2.5-7B-Instruct Qwen2.5-14B Qwen2.5-14B-Instruct Qwen2.5-32B Qwen2.5-32B-Instruct Qwen2.5-72B Qwen2.5-72B-Instruct	large language model	Build_In/PyTorch
QWQ	huggingface	model name QwQ-32B-Preview QwQ-32B	large language model	Build_In/PyTorch
Qwen3	huggingface	model name Qwen3-0.6B Qwen3-1.7B Qwen3-4B Qwen3-8B Qwen3-4B-Instruct-2507 Qwen3-4B-Thinking-2507	large language model	Build_In/PyTorch
DeepSeek-R1-Distill	huggingface	model name DeepSeek-R1-Distill-Qwen-1.5B DeepSeek-R1-Distill-Qwen-7B DeepSeek-R1-Distill-Qwen-14B DeepSeek-R1-Distill-Qwen-32B DeepSeek-R1-Distill-Llama-8B DeepSeek-R1-Distill-Llama-70B	large language model	Build_In/PyTorch

model	codebase	model list	model type	runtime
Qwen2.5	huggingface	model name Qwen2.5-7B-Instruct-GPTQ-Int4 Qwen2.5-14B-Instruct-GPTQ-Int4 Qwen2.5-32B-Instruct-GPTQ-Int4	large language model	vLLM
Qwen3	huggingface	model name Qwen3-30B-A3B-FP8 Qwen3-30B-A3B-Instruct-2507-FP8 Qwen3-30B-A3B-Thinking-2507-FP8	large language model	vLLM
DeepSeek-V3	huggingface	model name DeepSeek-V3-Base DeepSeek-V3 DeepSeek-V3-0324 DeepSeek-V3.1 DeepSeek-V3.1-Base	large language model	vLLM
DeepSeek-R1	huggingface	model name DeepSeek-R1 DeepSeek-R1-0528	large language model	vLLM

VLM Models

model	codebase	model list	model type	runtime
Qwen2-VL	huggingface	model name Qwen2-VL-2B-Instruct Qwen2-VL-7B-Instruct	vision-language model	Build_In/PyTorch
Qwen2.5-VL	huggingface	model name Qwen2.5-VL-3B-Instruct Qwen2.5-VL-7B-Instruct Qwen2.5-VL-32B-Instruct	vision-language model	Build_In/PyTorch

免责声明

• VastModelZOO提供的模型仅供您用于非商业目的，请参考原始模型来源许可证进行使用
• VastModelZOO描述的数据集均为开源数据集，如您使用这些数据集，请参考原始数据集来源许可证进行使用
• 如您不希望您的数据集或模型公布在VastModelZOO上，请您提交issue，我们将尽快处理

使用许可

• VastModelZOO提供的模型，如原始模型有许可证描述，请以该许可证为准
• VastModelZOO遵循Apache 2.0许可证许可