Models

Overview of machine learning Models in Gradient

The Gradient model repository is a hub for importing, managing, and deploying ML models.

Models are available within projects

Gradient Models can be created in two ways:

  1. Generate models from your workloads i.e. Workflows or Notebooks

  2. Import into Gradient by uploading them directly from the Web UI or CLI. Learn more here.

Gradient has a repository of Models per project. The model repository holds reference to the model artifacts (files generated during training) as well as optional summary metrics associated with the model's performance such as accuracy and loss.

Supported Model formats:

  • ONNX (Open Neural Network Exchange)

  • Custom