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 |