createTrainer
ファインチューニング学習を定義する。
createTrainer は学習を記述する場所です。どのベースモデルを使い、どのデータセットを使い、どんなパラメーターを回すか。返ってくる Trainer は arkor start(と Studio の Run training ボタン)が実行します。
import { createTrainer } from "arkor";
export const trainer = createTrainer({
name: "support-bot-v1",
model: "unsloth/gemma-4-E4B-it",
dataset: { type: "huggingface", name: "arkorlab/triage-demo" },
lora: { r: 16, alpha: 16 },
maxSteps: 100,
});最初に触るフィールド
name: Studio とクラウド側ログに表示。具体的な名前を。model: ベースのオープンウェイトモデル。テンプレートはgemma-4-E4B-itを使用。サポートしているモデル を参照。dataset: 学習データの所在。DatasetSourceを参照。lora: LoRA / QLoRA のパラメーター。r: 16, alpha: 16で十分。省略するとバックエンドのデフォルト。maxStepsまたはnumTrainEpochs: 学習時間を区切る。callbacks: コールバック を参照。
フル学習を走らせずに試す
dryRun: true でバックエンドにデータセットの切り詰めとステップ制限を指示し、パイプラインの全段階を走らせつつ数分で終わる形式に変えられます。コールバックを初めて組むときに便利です。
createTrainer({
name: "smoke",
model: "unsloth/gemma-4-E4B-it",
dataset: { type: "huggingface", name: "arkorlab/triage-demo" },
dryRun: true,
});リファレンス
TrainerInput の全フィールド、各型付きオプションフィールド、LoraConfig、未安定の forwarded フィールド(warmupSteps、loggingSteps、saveSteps、evalSteps など)、マルチトレーナーのロードマップは createTrainer リファレンス を参照してください。