Documentation

Supported models

πŸ“˜

Supported model lists

Model Compressor's compatibility is not limited to these model list.

The list shows validated environment and model examples to help you to start with.

You can easily get the model from torchvision.models, tf.keras.applications, or Model Compressor Model Zoo.


  • Supported version: PyTorch, ONNX version β‰₯ 1.10
  • If a model is defined in PyTorch, it should be converted into GraphModule or ONNX format before being uploaded.
  • How-to-guide for PyTorch GraphModule conversion is at the GitHub.
  • How-to-guide for ONNX conversion is at the GitHub.

Classification Models

ImageNet pre-trained models

ModelStructured PruningFilter DecompositionPyTorch & ONNX Version
AlexnetOO1.10.x
VGG16OO1.10.x
ResNet18OO1.10.x
SqueezeNet1_0OO1.10.x
DenseNet161OO1.10.x
MobileNet_v2OO1.10.x
MobileNet_v3_largeOO1.10.x
MobileNet_v3_smallOO1.10.x
Wide ResNet50_2OO1.10.x
MNASNet1_0OO1.10.x
EfficientNet_b0OO1.10.x
ResNext50_32x4dOO1.10.x
RegNet_y_400mfOO1.10.x
RegNet_x_400mfOO1.10.x

Semantic Segmentation Models

COCO2017 pre-trained models

ModelStructured PruningFilter DecompositionPyTorch & ONNX Version
FCN ResNet50OO1.10.x
FCN ResNet101OO1.10.x

TensorFlow-Keras - tf.keras.applications

  • Supported version: TensorFlow 2.3~2.8.
  • Custom layer must not be included in Keras H5 format (.h5).
  • The model must contain not only weights but also the structure of the model (do not use save_weights).
  • If there is a custom layer in the model, please upload with TensorFlow SavedModel format (.zip).

ImageNet pre-trained models

ModelStructured PruningFilter DecompositionTensorFlow Version
VGG16OO2.3.x ~ 2.8.x
VGG19OO2.3.x ~ 2.8.x
ResNet50OO2.3.x ~ 2.8.x
ResNet101OO2.3.x ~ 2.8.x
ResNet152OO2.3.x ~ 2.8.x
ResNet50V2OO2.3.x ~ 2.8.x
ResNet101V2OO2.3.x ~ 2.8.x
ResNet152V2OO2.3.x ~ 2.8.x
InceptionV3OO2.3.x ~ 2.8.x
MobileNetOO2.3.x ~ 2.8.x
MobileNetV2OO2.3.x ~ 2.8.x
DenseNet121OO2.3.x ~ 2.8.x
DenseNet169OO2.3.x ~ 2.8.x
DenseNet201OO2.3.x ~ 2.8.x
EfficientNetB1-B7OO2.3.x ~ 2.8.x
XceptionOO2.3.x ~ 2.8.x
InceptionResNetV2OX (WIP)2.3.x ~ 2.8.x
NASNetOX (WIP)2.3.x ~ 2.8.x

Classification Models

[PyTorch] CIFAR100 pre-trained models

ModelStructured PruningFilter DecompositionPyTorch & ONNX Version
MobileNetV2OO1.10.x
RepVGG-A1OO1.10.x
ResNet56OO1.10.x
VGG16-BNOO1.10.x

[TensorFlow-Keras] CIFAR100 pre-trained models

ModelStructured PruningFilter DecompositionTensorFlow Version
VGG19OO2.3.x ~ 2.8.x
ResNet50OO2.3.x ~ 2.8.x
MobileNetV1OO2.3.x ~ 2.8.x

Object Detection Models

[PyTorch] COCO pre-trained models

ModelStructured PruningFilter DecompositionPyTorch & ONNX Version
YOLOXOO1.10.x

[TensorFlow-Keras] Pascal VOC pre-trained models

ModelStructured PruningFilter DecompositionTensorFlow Version
YOLOv4OO2.3.x ~ 2.8.x