目前支持对包括但不限于表1中的模型进行模型训练后量化。
模式 |
类型 |
名称 |
|---|---|---|
Data-Free模式 |
图像分类 |
googleNet |
MobileNetV1 |
||
MobileNetV2 |
||
ResNet18 |
||
ResNet50 |
||
ResNet101 |
||
squeezeNet |
||
ShuffleNetV2 |
||
目标检测 |
Retinanet |
|
YoloV3 |
||
YoloV4 |
||
SSD |
||
Label-Free模式 |
图像分类 |
efficientNet |
efficientnet_B0 |
||
googleNet |
||
MobileNetV1 |
||
MobileNetV2 |
||
ResNet18 |
||
ResNet50 |
||
ResNet101 |
||
VGG16 |
||
VGG19 |
||
squeezeNet |
||
ShuffleNetV2 |
||
目标检测 |
fasterrcnn |
|
Retinanet |
||
YoloV3 |
||
YoloV4 |
||
SSD |
||
语义分割 |
UNet |