Skip to content

Latest commit

 

History

History
34 lines (31 loc) · 8.08 KB

CPU_Models_Containers.md

File metadata and controls

34 lines (31 loc) · 8.08 KB

Intel® CPU Model Containers

These are the deep learning Models optimized to run on Intel® Xeon® Scalable processors

Container Framework Docker Pull Command Compressed Size
ResNet 50v1.5 Training TensorFlow docker pull intel/image-recognition:centos-tf-cpu-resnet50v1-5-training 1.46GB
ResNet 50v1.5 Inference TensorFlow docker pull intel/image-recognition:centos-tf-cpu-resnet50v1-5-inference 1.69GB
BERT large Training TensorFlow docker pull intel/language-modeling:centos-tf-cpu-bert-large-pretraining 1.46GB
BERT large Inference TensorFlow docker pull intel/language-modeling:centos-tf-cpu-bert-large-inference 6.76GB
DIEN Inference TensorFlow docker pull intel/recommendation:centos-tf-cpu-dien-inference 1.61GB
MobileNet V1* Inference TensorFlow docker pull intel/image-recognition:centos-tf-cpu-mobilenet-v1-inference 1.39GB
3D U-Net MLPerf* Inference TensorFlow docker pull intel/image-segmentation:centos-tf-cpu-3d-unet-mlperf-inference 2.41GB
SSD-ResNet34 Training TensorFlow docker pull intel/object-detection:centos-tf-cpu-ssd-resnet34-training 2.83GB
SSD-ResNet34 Inference TensorFlow docker pull intel/object-detection:centos-tf-cpu-ssd-resnet34-inference 2.9GB
SSD-MobileNet* Inference TensorFlow docker pull intel/object-detection:centos-tf-cpu-ssd-mobilenet-inference 2.56GB
Transformer_LT_mlperf Training TensorFlow docker pull intel/language-translation:centos-tf-cpu-transformer-mlperf-training 1.46GB
Transformer_LT_mlperf Inference TensorFlow docker pull intel/language-translation:centos-tf-cpu-transformer-mlperf-inference 3.1GB
ResNet50 Training PyTorch docker pull intel/image-recognition:centos-pytorch-cpu-resnet50-training 1.63GB
ResNet50 Inference PyTorch docker pull intel/image-recognition:centos-pytorch-cpu-resnet50-inference 1.58GB
ResNext-32x16d Inference PyTorch docker pull intel/image-recognition:centos-pytorch-cpu-resnext-32x16d-inference 1.58GB
Mask R-CNN Training PyTorch docker pull intel/object-detection:centos-pytorch-cpu-maskrcnn-training 1.76GB
Mask R-CNN Inference PyTorch docker pull intel/object-detection:centos-pytorch-cpu-maskrcnn-inference 1.47GB
SSD-ResNet34 Training PyTorch docker pull intel/object-detection:centos-pytorch-cpu-ssd-resnet34-training 1.79GB
SSD-ResNet34 Inference PyTorch docker pull intel/object-detection:centos-pytorch-cpu-ssd-resnet34-inference 1.76GB
BERT large Training PyTorch docker pull intel/language-modeling:centos-pytorch-cpu-bert-large-training 1.99GB
BERT large Inference PyTorch docker pull intel/language-modeling:centos-pytorch-cpu-bert-large-inference 1.82GB
DistilBERT base Inference PyTorch docker pull intel/language-modeling:centos-pytorch-cpu-distilbert-inference 2.04GB
RNNT Training PyTorch docker pull intel/language-modeling:centos-pytorch-cpu-rnnt-training 1.9GB
RNNT Inference PyTorch docker pull intel/language-modeling:centos-pytorch-cpu-rnnt-inference 1.85GB
DLRM Training PyTorch docker pull intel/recommendation:centos-pytorch-cpu-dlrm-training 1.57GB
DLRM Inference PyTorch docker pull intel/recommendation:centos-pytorch-cpu-dlrm-inference 1.53GB

Intel AI Containers