Skip to content

Vastai/VastModelZOO

Repository files navigation

logo

为方便大家使用modelzoo,我们将持续增加典型网络和基础插件。如果您有任何需求,请提交issues,我们会及时处理。

model requirements

  • vaststream:编译器和推理引擎
  • vamc:模型转换工具

model summary

按算法方向主要从以下两个方面总结,该系列主要用来展示目前支持和将要支持的模型:

  • arch summary : 该算法方向的整体发展脉络和重点模型
  • op&module summary:该算法方向的算子清单和通用结构

model list

model codebase model list task type runtime
VGG timm
model name
  • vgg11
  • vgg11_bn
  • vgg13
  • vgg13_bn
  • vgg16
  • vgg16_bn
  • vgg19
  • vgg19_bn
classification E2E
VGG torchvision
model name
  • vgg11
  • vgg11_bn
  • vgg13
  • vgg13_bn
  • vgg16
  • vgg16_bn
  • vgg19
  • vgg19_bn
classification E2E
VGG mmcls
model name
  • vgg11
  • vgg11_bn
  • vgg13
  • vgg13_bn
  • vgg16
  • vgg16_bn
  • vgg19
  • vgg19_bn
classification E2E
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 E2E
ResNet torchvision
model name
  • resnet18
  • resnet34
  • resnet50
  • resnet101
  • resnet152
classification E2E
ResNet mmcls
model name
  • resnet18
  • resnet34
  • resnet50
  • resnet101
  • resnet152
classification E2E
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 E2E
ResNet keras
model name
  • resnet50
  • resnet50v2
  • resnet101
  • resnet101v2
  • resnet152
  • resnet152v2
classification E2E
ResNet oneflow
model name
  • resnet18
  • resnet34
  • resnet50
  • resnet101
  • resnet152
classification E2E
DenseNet timm
model name
  • densenet121
  • densenet161
  • densenet169
  • densenet201
  • densenetblur121d
classification E2E
DenseNet torchvision
model name
  • densenet121
  • densenet161
  • densenet169
  • densenet201
classification E2E
DenseNet mmcls
model name
  • densenet121
  • densenet161
  • densenet169
  • densenet201
classification E2E
DenseNet keras
model name
  • densenet121
  • densenet169
  • densenet201
classification E2E
DenseNet ppcls
model name
  • densenet121
  • densenet161
  • densenet169
  • densenet201
  • densenet264
classification E2E
MobileNet_v1 keras
model name
  • mobilenetv1
classification E2E
MobileNet_v1 ppcls
model name
  • mobilenet_v1_x0.25
  • mobilenet_v1_x0.5
  • mobilenet_v1_x0.75
  • mobilenet_v1
  • mobilenet_v1_ssld
classification E2E
MobileNet_v2 torchvision
model name
  • mobilenetv2
classification E2E
MobileNet_v2 timm
model name
  • mobilenetv2
classification E2E
MobileNet_v2 mmcls
model name
  • mobilenetv2
classification E2E
MobileNet_v2 keras
model name
  • mobilenetv2
classification E2E
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 E2E
MobileNet_v3 torchvision
model name
  • mobilenet_v3_small
  • mobilenet_v3_large
classification E2E
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 E2E
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 E2E
MobileNet_v3 showlo
model name
  • mobilenet_v3_small
classification E2E
ShuffleNet_v2 torchvision
model name
  • shufflenet_v2_x0_5
  • shufflenet_v2_x1_0
classification E2E
ShuffleNet_v2 mmcls
model name
  • shufflenet_v2_x1_0
classification E2E
ShuffleNet_v2 ppcls
model name
  • shufflenet_v2_x0_25
  • shufflenet_v2_x0_33
  • shufflenet_v2_x0_5
  • shufflenet_v2_x1_0
  • shufflenet_v2_x1_5
  • shufflenet_v2_x2_0
  • shufflenet_v2_swish
classification E2E
ShuffleNet_v2 megvii
model name
  • shufflenet_v2_x0.5
  • shufflenet_v2_x1.0
  • shufflenet_v2_x1.5
  • shufflenet_v2_x2.0
classification E2E
Xception timm
model name
  • xception
  • xception41
  • xception41p
  • xception65
  • xception65p
  • xception71
classification E2E
Xception keras
model name
  • xception
classification E2E
ResNeXt timm
model name
  • resnext50_32x4d
  • resnext50d_32x4d
  • resnext101_32x8d
  • resnext101_64x4d
  • tv_resnext50_32x4d
classification E2E
ResNeXt torchvision
model name
  • resnext50_32x4d
  • resnext101_32x8d
classification E2E
ResNeXt mmcls
model name
  • resnext50_32x4d
  • resnext101_32x4d
  • resnext101_32x8d
  • resnext152_32x4d
classification E2E
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 E2E
Wide ResNet timm
model name
  • wide_resnet50_2
  • wide_resnet101_2
classification E2E
Wide ResNet torchvision
model name
  • wide_resnet50_2
  • wide_resnet101_2
classification E2E
ResNeSt official
model name
  • resnest50
  • resnest101
  • resnest200
  • resnest269
classification E2E
HRNet official
model name
  • HRNet_w18_small_model_v1
  • HRNet_w18_small_model_v2
  • HRNet_w18
  • HRNet_w30
  • HRNet_w32
  • HRNet_w40
  • HRNet_w44
  • HRNet_w48
  • HRNet_w64
classification E2E
HRNet mmcls
model name
  • HRNet_w18
  • HRNet_w30
  • HRNet_w32
  • HRNet_w40
  • HRNet_w44
  • HRNet_w48
  • HRNet_w64
classification E2E
HRNet timm
model name
  • HRNet_w18_small_model_v1
  • HRNet_w18_small_model_v2
  • HRNet_w18
  • HRNet_w30
  • HRNet_w32
  • HRNet_w40
  • HRNet_w44
  • HRNet_w48
  • HRNet_w64
classification E2E
HRNet ppcls
model name
  • HRNet_W18_C
  • HRNet_W18_C_ssld
  • HRNet_W30_C
  • HRNet_W32_C
  • HRNet_W40_C
  • HRNet_W44_C
  • HRNet_W48_C
  • HRNet_W48_C_ssld
  • HRNet_W64_C
  • SE_HRNet_W64_C_ssld
classification E2E
Inception_v3 timm
model name
  • inception_v3
  • tf_inception_v3
  • adv_inception_v3
  • gluon_inception_v3
classification E2E
Inception_v3 torchvision
model name
  • inception_v3
classification E2E
Inception_v3 ppcls
model name
  • inception_v3
classification E2E
Inception_v4 timm
model name
  • inception v4
classification E2E
Xception ppcls
model name
  • xception41
  • xception41_deeplab
  • xception65
  • xception65_deeplab
  • xception71
classification E2E
CSPNet timm
model name
  • cspresnet50
  • cspresnext50
  • cspdarknet53
classification E2E
CSPNet mmcls
model name
  • cspresnet50
  • cspresnext50
  • cspdarknet53
classification E2E
CSPNet ppcls
model name
  • cspdarknet53
classification E2E
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 E2E
DLANet timm
model name
  • dla34
  • dla46_c
  • dla46x_c
  • dla60
  • dla60x
  • dla60x_c
  • dla102
  • dla102x
  • dla102x2
  • dla169
  • dla60_res2net
  • dla60_res2next
classification E2E
DLANet ppcls
model name
  • dla34
  • dla46_c
  • dla60
  • dla60x
  • dla60x_c
  • dla102
  • dla102x
  • dla102x2
  • dla169
classification E2E
DLANet ucbdrive
model name
  • dla34
  • dla46_c
  • dla46x_c
  • dla60
  • dla60x
  • dla60x_c
  • dla102
  • dla102x
  • dla102x2
  • dla169
classification E2E
VOVNet timm
model name
  • ese_vovnet19b_dw
  • ese_vovnet39b
classification E2E
MNasNet torchvision
model name
  • mnasnet0_5
  • mnasnet1_0
classification E2E
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 E2E
EfficientNet_v2 timm
model name
  • efficientnetv2_rw_t
  • efficientnetv2_rw_s
  • efficientnetv2_rw_m
classification E2E
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 E2E
SENet mmcls
model name
  • seresnet50
  • seresnet101
classification E2E
SENet ppcls
model name
  • seresnet18
  • seresnet34
  • seresnet50
  • senet154
  • seresnext50_32x4d
  • seresnext50_32x4d_vd
  • seresnext101_32x4d
classification E2E
GhostNet timm
model name
  • ghostnet_100
classification E2E
ESNet ppcls
model name
  • ESNet_x0_25
  • ESNet_x0_5
  • ESNet_x0_75
  • ESNet_x1_0
classification E2E
PPLCNet ppcls
model name
  • PPLCNet_x0_25
  • PPLCNet_x0_35
  • PPLCNet_x0_5
  • PPLCNet_x0_75
  • PPLCNet_x1_0
  • PPLCNet_x1_5
  • PPLCNet_x2_0
  • PPLCNet_x2_5
  • PPLCNet_x0_5_ssld
  • PPLCNet_x1_0_ssld
  • PPLCNet_x2_5_ssld
classification E2E
PPLCNet_v2 ppcls
model name
  • PPLCNetV2_base
  • PPLCNetV2_base_ssld
classification E2E
IResNet official
model name
  • iresnet50
  • iresnet101
  • iresnet152
  • iresnet200
classification E2E
Yolov5 ultralytics
model name
  • yolov5n
  • yolov5n6
  • yolov5s
  • yolov5s6
  • yolov5m
  • yolov5m6
  • yolov5l
  • yolov5l6
  • yolov5x
  • yolov5x6
object detection E2E
Bert google
model name
  • bert-base-mrpc
sentence classification E2E
Bert bert4torch
model name
  • bert-base-ner
named entity recognition E2E

Releases

No releases published

Packages

No packages published

Languages