Please visit this page for performance information.
This repository is a collection of models that have been ported to run on Habana Gaudi training accelerators. They are intended as examples, and will be reasonably optimized for performance while still being easy to read.
| Models | Framework |
|---|---|
| ResNet 50 Keras | TensorFlow |
| ResNeXt 50, 101, 152 | TensorFlow |
| SSD | TensorFlow |
| Mask R-CNN | TensorFlow |
| DenseNet | TensorFlow |
| UNet 2D | TensorFlow |
| UNet 3D | TensorFlow |
| CycleGAN | TensorFlow |
| EfficientDet | TensorFlow |
| RetinaNet | TensorFlow |
| SegNet | TensorFlow |
| MobileNet V2 | TensorFlow |
| ResNet50, ResNet152, ResNext101 | PyTorch |
| UNet 2D | PyTorch |
| Models | Framework |
|---|---|
| BERT | TensorFlow |
| ALBERT | TensorFlow |
| Transformer | TensorFlow |
| T5 Base | TensorFlow |
| BERT | PyTorch |
| RoBERTa | PyTorch |
| DistilBERT | PyTorch |
| Transformer | PyTorch |
| Models | Framework |
|---|---|
| DLRM | PyTorch |
We welcome you to use the GitHub issue tracker to report bugs or suggest features.
When filing an issue, please check existing open, or recently closed, issues to make sure somebody else hasn't already reported the issue. Please try to include as much information as you can. Details like these are incredibly useful:
- A reproducible test case or series of steps
- The version of our code being used
- Any modifications you've made relevant to the bug
- Anything unusual about your environment or deployment