Model training

Train custom vision models with no infrastructure to manage.

Pick a task and a GPU. Pictograph exports your data, trains a custom model on managed GPUs, and returns the weights.

train.py

from pictograph import Client
from pictograph.pipelines import train_pipeline

client = Client()

# Export the dataset, train on a managed GPU, fetch the model
run, model = train_pipeline(
    client,
    "road-signs",
    pipeline="yolox",   # object detection
    gpu="a10g",         # a10g / a100
)

# Download the trained model weights
client.models.download(model.id, "road_signs.onnx")

How it works

From dataset to trained model

No infrastructure to manage. Pictograph converts your dataset to the right format, runs the job, and returns model weights.

Build a dataset

Annotate images in the editor, auto-annotate with SAM3, or import from V7 or Roboflow.
Pick an architecture and GPU

Choose a vision task and a managed GPU. Pictograph handles the data conversion for you.
Train on managed GPUs

Your job runs on managed GPUs with live progress. No Colab limits, no GPU setup.
Get model weights

Receive trained model weights in your registry. Deploy as an endpoint in one click.

Pick your architecture

Five vision tasks, one trained model

Choose the architecture that fits your task. Every pipeline returns model weights, ready to deploy as an endpoint or run in a workflow.

YOLOX Object detection. Fast and edge-deployable.

RF-DETR Strong detection and instance segmentation with pixel masks.

Segformer Semantic segmentation, pixel-wise class maps.

Classification Whole-image labels.

Train from code

Kick off a run from the SDK or the CLI

The train_pipeline workflow chains export, training, and model fetch into one call. Or run it from the command line.

Python SDK

train.py

from pictograph import Client
from pictograph.pipelines import train_pipeline

client = Client()

# Export the dataset, train on a managed GPU, fetch the model
run, model = train_pipeline(
    client,
    "road-signs",
    pipeline="yolox",   # object detection
    gpu="a10g",         # a10g / a100
)

# Download the trained model weights
client.models.download(model.id, "road_signs.onnx")

CLI

cli

pictograph train \
  --dataset ds_road_signs \
  --pipeline yolox \
  --gpu a10g

# Watch progress, then download the model weights
pictograph models download mdl_abc123 road_signs.onnx

Managed GPUs, transparent pricing. Training runs on managed A10G and A100 GPUs. You pay in transparent USD compute credits priced per GPU-minute, with no infrastructure to provision.

Keep exploring

Auto-annotation Label a dataset fast with SAM3 point, box, and text prompts before you train. Deployments Stand up your trained model as an always-on /predict endpoint in one click. Workflows Run your model over images, video, or a dataset in a visual pipeline.

FAQ

Model training FAQ

What models can I train on Pictograph?

You can train object detection (YOLOX, RF-DETR), instance segmentation (RF-DETR), semantic segmentation (SM-PyTorch), and image classification models. Every model exports to ONNX for portable inference.

How much data do I need to train a model?

A minimum of five annotated images is required for the train, validation, and test split, but more labeled data improves accuracy. Use SAM3 auto-annotation to label a dataset quickly before training.

How is training priced?

Training is billed by GPU time on managed A10G GPUs at the underlying rate plus the platform markup, charged from compute credits. You pay only for the actual training run, with no idle cost.

What happens after training finishes?

The trained model is exported to ONNX and added to your model registry with its evaluation metrics. From there you can run batch inference in a workflow, stand it up as a live deployment, or publish it publicly.

Train your first model free

$5/mo free compute. No payment method required.

Start free Read the docs