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
Wide & Deep
Wide & Deep Recommender Model Training For TensorFlow 2
Unet2D
UNet for 2D Medical Image Segmentation for TensorFlow
Transformer
Transformer for TensorFlow
SSD ResNet34
Single Shot Detection (SSD) for TensorFlow