diff --git a/CLAUDE.md b/CLAUDE.md index 2f077ab..3038d30 100644 --- a/CLAUDE.md +++ b/CLAUDE.md @@ -129,9 +129,9 @@ Baseline 数据:推理 229s,AUC 0.759,PCOC 1.110,得分 25.85。 1. ✅ **接口对齐** — 确认能在评测系统跑通(得分 > 0) 2. ✅ **FP16 量化** — `model.half()`,Embedding 保留 FP32,152s 3. ✅ **Flash Attention** — 替换 `scaled_dot_product` 为 `F.scaled_dot_product_attention`,94.5s -4. 🔲 **torch.compile(default)** — 纯算子融合,不用 CUDA Graph,待验证(与 inference_mode 同包提交) -5. ✅ **inference_mode()** — 替代 `no_grad()`,关闭版本追踪 -6. ❌ **MoE Top-1 gating** — 已回退:PCOC 从 1.059 炸到 2.075,输出校准崩坏 +4. ✅ **inference_mode()** — 替代 `no_grad()`,92.5s(+2s 小幅提升) +5. ❌ **torch.compile** — reduce-overhead 和 default 模式均因动态 batch 形状反效果,彻底放弃 +6. ❌ **MoE Top-1 gating** — PCOC 从 1.059 炸到 2.075,已回退 7. 🔲 **2:4 结构化稀疏** — A800 原生加速,权重形状不变(显式允许) CUDA Graph 已评估并放弃(batch 形状不固定,不适用)。 @@ -155,6 +155,8 @@ CUDA Graph 已评估并放弃(batch 形状不固定,不适用)。 | 日期 | 提交次数 | 得分 | AUC | PCOC | 耗时 | 优化手段 | 备注 | |------|----------|------|-----|------|------|----------|------| -| 06/12 | 8 | **0** | 0.736 | 2.075 | 119.6s | MoE k=1 + compile(default) | PCOC 炸毁,已回退 MoE | -| 06/12 | 7 | **56.98** | 0.7526 | 1.059 | 94.5s | + Flash Attention | build_env.sh 预热脚本导致第5次异常 | -| 06/12 | 3 | **43.55** | 0.7525 | 1.059 | 152s | 接口对齐 + FP16 量化 | 第 1、2 次因 requirements.txt 异常 | +| 06/13 | 10 | **57.45** | 0.7526 | 1.059 | 92.5s | + inference_mode | **当前最优** | +| 06/13 | 9 | 51.42 | 0.7525 | 1.059 | 118.4s | + compile(default) | 反效果,已移除 | +| 06/12 | 8 | 0 | 0.736 | 2.075 | 119.6s | MoE k=1 + compile | PCOC 炸毁 | +| 06/12 | 6 | 56.98 | 0.7526 | 1.059 | 94.5s | + Flash Attention | | +| 06/12 | 3 | 43.55 | 0.7525 | 1.059 | 152s | + FP16 量化 | |