Trackio documentation

TRL Entegrasyonu

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

TRL Entegrasyonu

Trackio, TRL ile yerel olarak entegre edilmiştir; böylece herhangi bir TRL eğiticisinden (SFTTrainer, DPOTrainer, GRPOTrainer vb.) metrikleri en az kurulumla kaydedebilirsiniz. trl kütüphanesinin en son sürümünün (1.2.0 veya üzeri) yüklü olduğundan emin olun.

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()

Proje ve Uzay Yapılandırması

Proje ve alan kimliğini doğrudan TRL yapılandırmanızda ayarlayın (ör. 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