Sign in

v4

latestOpenAPI 3.1.0MIT2026-08-04193459793.6 KB
RL

Create training session

Creates a training session and returns its details.

post/rl/training-sessions

Request body

resume_from_checkpoint_idstring

Checkpoint ID to resume from

resume_from_hf_checkpointstring

HuggingFace repo (or hf://) to resume model weights from. Accepts either a full model or a PEFT adapter directory. Mutually exclusive with resume_from_checkpoint_id.

model_resources_idstring required

Model resource to attach the session to. The session runs on that resource's GPU pods.

display_namestring

Optional display name used to identify the training session

Example request

{
  "resume_from_checkpoint_id": "123e4567-e89b-12d3-a456-426614174000",
  "resume_from_hf_checkpoint": "your-org/llama-3-8b-finetuned",
  "lora_config": {
    "rank": 32,
    "alpha": 64
  },
  "model_resources_id": "123e4567-e89b-12d3-a456-426614174000",
  "display_name": "gsm8k-experiment-2",
  "metadata": {
    "wandb": {
      "entity": "example-org",
      "project": "grpo-gsm8k",
      "group": "gsm8k-35b-sweep",
      "run_name": "exp2-thinking-4k-ctx",
      "run_id": "abc123",
      "url": "https://wandb.ai/example-org/example-project/runs/run-id"
    }
  }
}

Response

Training session details

idstring required

ID of the training session

display_namestring

Display name used to identify the training session

status'TRAINING_SESSION_STATUS_UNSPECIFIED' | 'TRAINING_SESSION_STATUS_CREATING' | 'TRAINING_SESSION_STATUS_RUNNING' | 'TRAINING_SESSION_STATUS_STOPPED' | 'TRAINING_SESSION_STATUS_STOPPING' | 'TRAINING_SESSION_STATUS_ERROR' | 'TRAINING_SESSION_STATUS_EXPIRED' required

Status of the training session

resume_from_checkpoint_idstring

Checkpoint ID this session was resumed from

created_atstring date-time required

Timestamp when the training session was created

updated_atstring date-time required

Timestamp when the training session was last updated

model_resources_idstring required

Model resource this session is attached to. The session runs on that resource's GPU pods.

created_bystring required

ID of the user who created the training session

Example response

{
  "id": "123e4567-e89b-12d3-a456-426614174000",
  "display_name": "gsm8k-experiment-2",
  "metadata": {
    "wandb": {
      "entity": "example-org",
      "project": "grpo-gsm8k",
      "group": "gsm8k-35b-sweep",
      "run_name": "exp2-thinking-4k-ctx",
      "run_id": "abc123",
      "url": "https://wandb.ai/example-org/example-project/runs/run-id"
    }
  },
  "error": {
    "message": "The session could not start because the selected resource has no available capacity.",
    "occurred_at": "2026-01-02T00:00:05Z"
  },
  "inference_checkpoints": [
    {
      "id": "123e4567-e89b-12d3-a456-426614174000",
      "created_at": "2026-01-02T00:00:00Z",
      "registration": {
        "model_name": "username/Meta-Llama-3-8B-rl-step-42-20260216",
        "registered_at": "2026-01-02T00:00:00Z"
      }
    }
  ],
  "training_checkpoints": [
    {
      "id": "123e4567-e89b-12d3-a456-426614174000",
      "created_at": "2026-01-02T00:00:00Z"
    }
  ],
  "resume_from_checkpoint_id": "123e4567-e89b-12d3-a456-426614174000",
  "created_at": "2026-01-02T00:00:00Z",
  "updated_at": "2026-01-02T00:00:05Z",
  "lora_config": {
    "rank": 32,
    "alpha": 64
  },
  "model_resources_id": "123e4567-e89b-12d3-a456-426614174000",
  "created_by": "user_123"
}