Trackio

Integración con Transformers

Trackio se integra de forma nativa con Transformers para que puedas registrar métricas con una configuración mínima. Asegúrate de tener instalada la última versión de transformers (versión 4.54.0 o superior).

import numpy as np
from datasets import Dataset
from transformers import Trainer, AutoModelForCausalLM, TrainingArguments

# Create a fake dataset
data = np.random.randint(0, 1000, (8192, 64)).tolist()
dataset = Dataset.from_dict({"input_ids": data, "labels": data})

# Train a model using the Trainer API
trainer = Trainer(
    model=AutoModelForCausalLM.from_pretrained("Qwen/Qwen3-0.6B"),
    args=TrainingArguments(report_to="trackio", run_name="Qwen3-0.6B-training"),
    train_dataset=dataset,
)
trainer.train()

Configuración del proyecto y el espacio

Establece el ID del proyecto y del espacio directamente en tus TrainingArguments:

from transformers import TrainingArguments

args = TrainingArguments(
    report_to="trackio",
    run_name="my-run",
    project="my-project",
    trackio_space_id="username/space_id",
)

Update on GitHub