Agent 信息
执行 Agent:Kimi(Moonshot AI),运行在 prime-agent 会话中。
Kimi 通过 prime-agent 的工具链完成本次流水线:编写和更新 Kaggle kernel、调用 Kaggle GPU 训练、读取远端产物、使用 Hugging Face 工具合并模型并上传仓库。
Agent 自主完成的真实流水线
这个模型不是手工跑出来的结果。整个流程由 Agent 工具链完成:
-
生成/更新 Kaggle kernel
- 用 Kaggle API / CLI 提交 notebook job
- 让 kernel 在 Kaggle GPU runner 上自动执行训练
-
在 Kaggle 调用免费 GPU 算力
- GPU:
NVIDIA Tesla T4
- 由于 T4 是
sm_75,不支持 bf16,训练参数改为 fp16=True, bf16=False
- 完成 SFT 训练并保存 adapter
-
远程取回训练产物
- 用
kaggle kernels output 拉取训练结果
- 产物包括 LoRA adapter、tokenizer、日志和验证报告
-
使用 Hugging Face 工具链
- 本机用
peft / transformers 把 LoRA merge 到基座
- 生成
merged/ 完整模型目录
- 用
huggingface_hub.HfApi 上传模型到 Hugging Face Hub
-
整理仓库为可直接使用
adapter/:LoRA adapter
merged/:已经 merge 完成的完整模型,可直接 from_pretrained
- 清理 tokenizer 兼容问题,确保 transformers 4.x 可读
-
验证与更新 README
- 检查训练日志和 probe 结果
- teacher-forced stop probe 已验证:
recall_compound: rank 0
in_domain_inflation: rank 0
- 用最终信息更新这份 README
这条链路里,Kaggle 负责执行和计算,Hugging Face 负责模型托管与分发,本地只承担 assemblage、merge、验证和上传。
结果快照
- train_n = 8000, eval_n = 800
- train_loss = 1.5185747299194337
- eval_loss = 1.5003085136413574
- eos_token_id = 151643(
<|im_end|>)
推荐使用:直接加载 merged
from transformers import AutoTokenizer, AutoModelForCausalLM tok = AutoTokenizer.from_pretrained( "zhengr/mixtao-zh-sft-0.5b-lora", subfolder="merged", use_fast=False,)m = AutoModelForCausalLM.from_pretrained( "zhengr/mixtao-zh-sft-0.5b-lora", subfolder="merged", torch_dtype="auto", device_map="auto",)
仓库布局
adapter/ LoRA adaptermerged/ 已融合完整模型,可直接加载
说明
merged/tokenizer_config.json 已整理为兼容格式。
- 当前 README 记录的是最终可交付状态;过程验证以 Kaggle kernel 输出与本机验证结果为准。
- full greedy generation 仍可继续跑一次作为最终补充验收。
仓库目录
Table with columns: 目录, 内容, 配置| 目录 | 内容 | 配置 |
|---|
| 根目录 | 早期 LoRA(v9,基座 Qwen/Qwen2.5-0.5B) | r=32,全模块 |
merged/ | v14 全模型(基座 Qwen/Qwen2.5-0.5B-Instruct) | r=32,全模块,1 epoch,lr 2e-4 |
adapter-v15/ | v15 LoRA(基座 Qwen/Qwen2.5-0.5B-Instruct) | r=16/alpha=32,只挂 q_proj+v_proj,2 epoch,lr 2e-5 + warmup_steps 50 |
merged-v15/ | v15 全模型,可直接 from_pretrained | 同上,已 merge |
run-v15/ | v15 的 DONE.json 与训练 notebook | — |
v15 是按 Hugging Face 官方 huggingface/skills 里的 huggingface-llm-trainer / huggingface-datasets 两个 skill 的流程重跑的一版:开跑前先用 Dataset Viewer API 验证数据集子集与列名,用 Trackio 做训练监控,Hub 操作走 hf CLI。算力仍在 Kaggle T4(HF Jobs 需要付费计划)。
加载 v15
from transformers import AutoModelForCausalLM, AutoTokenizerR = "zhengr/mixtao-zh-sft-0.5b-lora"tok = AutoTokenizer.from_pretrained(R, subfolder="merged-v15")model = AutoModelForCausalLM.from_pretrained(R, subfolder="merged-v15")
v15 指标
Table with columns: 指标, 值| 指标 | 值 |
|---|
| train_n / eval_n | 8800 / 800 |
| train_loss | 1.6062 |
| eval_loss | 1.5649 |
| epochs | 2 |
| train_runtime | 1362 s |
停止行为探针(教师强制,取答案末尾下一 token 的 logits 排名):
Table with columns: 探针, 基座 rank, 训练后 rank| 探针 | 基座 rank | 训练后 rank |
|---|
| 续写历史(复利) | 0 | 0(stop_logit 21.48,top1 <|im_end|>) |
| 域内(通胀与资产配置) | 0 | 601(stop_logit 9.08,top1「例如」) |
greedy 生成抽测:两题均在 211 / 204 token 处干净停在 <\|im_end\|>。