Training scripts

The Viam registry hosts custom Python ML training scripts, which you can use to train machine learning models. You can upload your own training script by following the guide to Create custom training scripts.

You can search the available ML training scripts from the Viam registry here:

Script
Description

API

To submit training jobs programmatically, use the ML Training client API, which supports the following methods:

Method NameDescription
SubmitTrainingJobSubmit a training job.
SubmitCustomTrainingJobSubmit a training job from a custom training script.
GetTrainingJobGet training job metadata.
ListTrainingJobsGet training job metadata for all jobs within an organization.
CancelTrainingJobCancel the specified training job.
DeleteCompletedTrainingJobDelete a completed training job from the database, whether the job succeeded or failed.

Next steps

Follow one of these guides to write your own custom training script or to train models with training scripts:

Train computer vision models
Train computer vision models

Use your image data to create and label a dataset and train a computer vision ML model.

Create custom training scripts
Create custom training scripts

How to write and upload a custom machine learning model training script to the Viam registry and use it to submit training jobs.