Training scripts

The Viam registry hosts custom Python ML training scripts, which you can use to train machine learning models.

Training scripts in the registry

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 and deploy computer vision models
Train and deploy computer vision models

Use your image data to train a model, so your machines can make inferences about their environments.

Train models with custom specs
Train models with custom specs

If you want to train models to custom specifications, write a custom training script and upload it to the Viam registry.