zDLC

ONNX models

Obtaining the models

The ONNX models for the examples can be found at ONNX Model Zoo. A git clone of the repository is not required. Instead, follow the git-lfs instructions in the ONNX Model Zoo README.md to download a single model.

Example models

The following models are used in the examples:

  1. mnist-12

Verified ONNX Model Zoo models

The following list of ONNX models from the ONNX Model Zoo have been built and verified with the IBM Z Deep Learning Compiler:

Model Group Model
Alexnet bvlcalexnet-6
  bvlcalexnet-7
  bvlcalexnet-8
  bvlcalexnet-9
  bvlcalexnet-12
Bert-squad bertsquad-10
  bertsquad-12
Caffenet caffenet-6
  caffenet-7
  caffenet-8
  caffenet-9
  caffenet-12
Densenet-121 densenet-6
  densenet-7
  densenet-8
  densenet-9
  densenet-12
Efficientnet-lite4 efficientnet-lite4-11
Fast_neural_style candy-8
  candy-9
  mosaic-8
  mosaic-9
  pointilism-8
  pointilism-9
  rain-princess-8
  rain-princess-9
  udnie-8
  udnie-9
Googlenet googlenet-3
  googlenet-6
  googlenet-7
  googlenet-8
  googlenet-9
  googlenet-12
Gpt-2 gpt2-10
  gpt2-lm-head-10
Inception_v1 inception-v1-6
  inception-v1-7
  inception-v1-8
  inception-v1-9
  inception-v1-12
Inception_v2 inception-v2-7
  inception-v2-8
  inception-v2-9
Mnist mnist-7
  mnist-8
  mnist-12
Mobilenet mobilenetv2-7
  mobilenetv2-10
Rcnn_ilsvrc13 rcnn-ilsvrc13-6
  rcnn-ilsvrc13-7
  rcnn-ilsvrc13-8
  rcnn-ilsvrc13-9
Resnet resnet18-v1-7
  resnet18-v2-7
  resnet34-v1-7
  resnet34-v2-7
  resnet50-caffe2-v1-6
  resnet50-caffe2-v1-7
  resnet50-caffe2-v1-8
  resnet50-caffe2-v1-9
  resnet50-v1-7
  resnet50-v1-12
  resnet50-v2-7
  resnet101-v1-7
  resnet101-v2-7
  resnet152-v1-7
  resnet152-v2-7
Roberta roberta-base-11
  roberta-sequence-classification-9
Shufflenet shufflenet-6
  shufflenet-7
  shufflenet-8
  shufflenet-9
  shufflenet-v2-10
  shufflenet-v2-12
Squeezenet squeezenet1.0-3
  squeezenet1.0-6
  squeezenet1.0-7
  squeezenet1.0-8
  squeezenet1.0-9
  squeezenet1.0-12
  squeezenet1.1-7
T5 t5-encoder-12
  t5-decoder-with-lm-head-12
Tiny-yolov2 tinyyolov2-7
  tinyyolov2-8
Tiny-yolov3 tiny-yolov3-11
Vgg vgg16-7
  vgg16-12
  vgg16-bn-7
  vgg19-7
  vgg19-bn-7
  vgg19-caffe2-6
  vgg19-caffe2-7
  vgg19-caffe2-8
  vgg19-caffe2-9
Yolov2-coco yolov2-coco-9
Yolov3 yolov3-10
  yolov3-12
Zfnet-512 zfnet512-6
  zfnet512-7
  zfnet512-8
  zfnet512-9
  zfnet512-12