This repository provides examples of deep learning models enabled with Habana(R) Gaudi(R) accelerators.
For more information about training deep learning models on Gaudi, visit developer.habana.ai.
Models | Framework |
---|---|
ResNets (ResNet50/101/152, ResNext101) | TensorFlow |
SSD | TensorFlow |
Mask R-CNN | TensorFlow |
DenseNet | TensorFlow |
ResNet-50 | PyTorch |
Models | Framework |
---|---|
BERT | TensorFlow |
BERT | PyTorch |
Models | Framework |
---|---|
DLRM | PyTorch |