Computer vision

Important

AI Runtime for single-node tasks is in Public Preview. The distributed training API for multi-GPU workloads remain in Beta.

This page provides notebook examples for computer vision tasks using AI Runtime. These examples demonstrate how to train and fine-tune models for various computer vision applications.

Tutorial	Description
Image classification using convolutional neural network	This notebook provides a simple example of how to train a 2-D convolution neural network on serverless GPUs for image classification.
Object detection using RetinaNet	This notebook demonstrates how to train an object detection model using RetinaNet on serverless GPU.