/Model-References

TensorFlow and PyTorch Reference models for Gaudi(R)

Primary LanguageJupyter Notebook

Habana Deep Learning Examples for Training

Introduction

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.

Computer Vision

Models Framework
ResNets (ResNet50/101/152, ResNext101) TensorFlow
SSD TensorFlow
Mask R-CNN TensorFlow
DenseNet TensorFlow
ResNet-50 PyTorch

Natural Language Processing

Models Framework
BERT TensorFlow
BERT PyTorch

Recommender Systems

Models Framework
DLRM PyTorch