Trackio documentation
Transformers 통합
Transformers 통합
Trackio는 Transformers와 네이티브로 통합되어 있어 최소한의 설정으로 메트릭을 기록할 수 있습니다. transformers의 최신 버전(4.54.0 이상)이 설치되어 있는지 확인하세요.
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()프로젝트 및 Space 구성
TrainingArguments에 프로젝트 및 스페이스 ID를 직접 설정하세요:
from transformers import TrainingArguments
args = TrainingArguments(
report_to="trackio",
run_name="my-run",
project="my-project",
trackio_space_id="username/space_id",
)