Trackio

Tích hợp TRL

Trackio tích hợp sẵn với TRL để bạn có thể ghi lại các chỉ số từ bất kỳ trình huấn luyện TRL nào (SFTTrainer, DPOTrainer, GRPOTrainer, v.v.) với cấu hình tối thiểu. Hãy đảm bảo bạn đã cài đặt phiên bản mới nhất của trl (phiên bản 1.2.0 trở lên).

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

Cấu hình Dự án và Space

Đặt trực tiếp ID dự án và ID space trong cấu hình TRL của bạn (ví dụ: 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