Views
No views yet
仅含模型权重（model.safetensors+metadata.pt）。不含训练优化器状态optimizer.pt。
| 路径 | 说明 |
|---|---|
adv-flatten-fold-run16/model.safetensors | 模型权重（~7.5 GB，PyTorch safetensors） |
adv-flatten-fold-run16/metadata.pt | 训练元信息（config / exp_name / 训练步数 / 超参数） |
训练时保存的optimizer.pt（~13.5 GB）为优化器状态，未随本仓库发布，对推理无影响。
| 项 | 值 |
|---|---|
| 模型类型 | Advantage Estimator（π₀.₅ 变体，仅保留价值/进度回归头） |
| 基于 | π₀.₅ 基础 checkpoint（见 metadata.pt / pytorch_weight_path） |
| 训练步数 | 20000 |
| 训练损失 | loss_value_weight=1.0，loss_action_weight=0.0（禁用动作损失） |
| 归一化 | skip_norm_stats=True（价值模型不依赖 norm_stats） |
| 输出 | 每帧 absolute_value（0~1 累计进度） |
| Config | ADVANTAGE_TORCH_KAI0_FLATTEN_FOLD_FULL_FINETUNE_MIX_02 |
| exp_name | full_finetune_460_run_16 |
stage_advantage/annotation/gt_label.py 离散化为正/负标签
写入 task_index / tasks.jsonl，再用于 Advantage-Weighted Behavior Cloning
（prompt_from_task=True）。absolute_value（0~1 累计进度）可把任务切成 K 个阶段，用于 subtask 级别成功判定与评估。ckpt_dir 指向本权重目录、ckpt_steps=20000（编辑
stage_advantage/annotation/eval.py 的 MODELS_CONFIG_MAP），然后：uv run python stage_advantage/annotation/eval.py Task-A KAI0 /path/to/dataset| 指标 | 值 |
|---|---|
| Frame Accuracy | 51.0%（±12.2%） |
| Stage IoU | 48.1% |
| Boundary MAE | 60.1 帧（~2.0 s） |
| Boundary Acc @±30f | 66.8% |
| Boundary Acc @±50f | 77.7% |
adv-flatten-fold-run16/（本仓库，推荐）：full_finetune_460_run_16，报告所用。run_15（full_finetune_460_run_15）权重不同，暂未随本仓库发布，请勿混用。LICENSE_GEMMA.txt（Gemma Terms of Use）OpenDriveLab-org/Kai0