ELECTRA Fine Tuning
ELECTRA Fine Tuning For TensorFlow2
SSD ResNet34
Single Shot Detection (SSD) for TensorFlow
TensorFlow
These are the available TensorFlow Container images in the Habana Vault.
These images contain SynapseAI SW and TensorFlow framework that allow a user to run models on Gaudi.
For more details on docker usage, please refer to our Installation Guide on instructions to setup Container Runtime and Installation of docker images
Transformer
Transformer for TensorFlow
Unet2D
UNet for 2D Medical Image Segmentation for TensorFlow
VGG-Segnet
VGG-Segnet for TensorFlow
Wide & Deep
Wide & Deep Recommender Model Training For TensorFlow 2