Trackio

Tích hợp Transformers

Trackio tích hợp sẵn với Transformers để bạn có thể ghi lại các chỉ số với cấu hình tối thiểu. Đảm bảo bạn đã cài đặt phiên bản mới nhất của transformers (phiên bản 4.54.0 trở lên).

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

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

Đặt ID dự án và không gian trực tiếp trong TrainingArguments của bạn:

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