Skip to main content

Computer vision

Beta

This feature is in Beta.

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

Image classification using convolutional neural network

The following notebook provides a simple example of how to train a 2-D convolution neural network on Serverless GPU compute for image classification.

PyTorch image classification

Open notebook in new tab