Skip to main content

Computer vision

Public Preview

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.