Trackio

Integración de TRL

Trackio se integra de forma nativa con TRL para que puedas registrar métricas de cualquier entrenador de TRL (SFTTrainer, DPOTrainer, GRPOTrainer, etc.) con una configuración mínima. Asegúrate de tener instalada la última versión de trl (versión 1.2.0 o superior).

from datasets import Dataset
from trl import SFTConfig, SFTTrainer

# Create a small fake dataset
prompts = ["The capital of France is", "Hamlet was written by"] * 12
completions = [" Paris.", " Shakespeare."] * 12
dataset = Dataset.from_dict({"prompt": prompts, "completion": completions})

# Train a model using the TRL SFTTrainer API
trainer = SFTTrainer(
    model="Qwen/Qwen3-0.6B",
    args=SFTConfig(report_to="trackio", run_name="Qwen3-0.6B-sft"),
    train_dataset=dataset,
)
trainer.train()

Configuración del proyecto y el espacio

Establece el ID del proyecto y del espacio directamente en tu configuración de TRL (p. ej. SFTConfig, DPOConfig, GRPOConfig):

from trl import SFTConfig

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

Update on GitHub