Tích hợp Trackio

Trackio là thư viện theo dõi thí nghiệm nhẹ, miễn phí, được xây dựng dựa trên 🤗 Datasets và 🤗 Spaces. Đây là giải pháp theo dõi được khuyến nghị cho TRL và được tích hợp sẵn với tất cả các trainer.

Để bật ghi log, chỉ cần đặt report_to="trackio" trong cấu hình huấn luyện của bạn:

from trl import SFTConfig  # works with any trainer config (e.g. DPOConfig, GRPOConfig, etc.)

training_args = SFTConfig(
    ...,
    report_to="trackio",  # enable Trackio logging
)

Tổ chức các thí nghiệm của bạn bằng tên chạy và dự án

Mặc định, Trackio sẽ tạo một tên để nhận diện mỗi lần chạy. Tuy nhiên, chúng tôi rất khuyến khích bạn đặt một run_name mô tả rõ ràng để dễ dàng tổ chức các thí nghiệm hơn. Ví dụ:

from trl import SFTConfig

training_args = SFTConfig(
    ...,
    report_to="trackio",
    run_name="sft_qwen3-4b_lr2e-5_bs128",  # descriptive run name
)

Bạn cũng có thể nhóm các thí nghiệm liên quan theo dự án bằng cách đặt biến môi trường sau:

export TRACKIO_PROJECT="my_project"

Lưu trữ Nhật ký của bạn trên 🤗 Spaces

Trackio có thiết kế ưu tiên cục bộ, nghĩa là các nhật ký của bạn được lưu trên máy của bạn. Nếu bạn muốn lưu trữ chúng và triển khai một bảng điều khiển trên 🤗 Spaces, hãy đặt:

export TRACKIO_SPACE_ID="username/space_id"

Chạy ví dụ sau:

import os
from trl import SFTConfig, SFTTrainer
from datasets import load_dataset

os.environ["TRACKIO_SPACE_ID"] = "trl-lib/trackio"
os.environ["TRACKIO_PROJECT"] = "trl-documentation"

trainer = SFTTrainer(
    model="Qwen/Qwen3-0.6B",
    train_dataset=load_dataset("trl-lib/Capybara", split="train"),
    args=SFTConfig(
        report_to="trackio",
        run_name="sft_qwen3-0.6b_capybara",
    ),
)
trainer.train()

sẽ cung cấp cho bạn một bảng điều khiển được lưu trữ tại https://huggingface.co/spaces/trl-lib/trackio.