add executable pi0.5 reproduction notebook

This commit is contained in:
seeker
2026-07-12 21:22:25 +08:00
parent b57516b57b
commit 7e0267be1f
4 changed files with 1197 additions and 0 deletions
+1
View File
@@ -5,5 +5,6 @@ __pycache__/
*.egg-info/
.venv/
data/
notebook_artifacts/
outputs/**/*.pt
.DS_Store
+7
View File
@@ -72,6 +72,13 @@ python -m pip install -e '.[dev]'
如果该路径不同,请先修改 `configs/overfit.yaml` 的 `data.source_manifest`。
## 一键 Notebook 复现
[`notebooks/pi05_minimal_reproduction.ipynb`](notebooks/pi05_minimal_reproduction.ipynb)
提供从环境安装、论文与架构说明、数据/权重导入、测试、完整训练、推理到结果图的
顺序执行流程。Notebook 支持本地 manifest、数据 zip URL 或自动生成的确定性最小
样本;默认从零训练,因此不要求预训练权重。
## 运行测试
```bash
File diff suppressed because one or more lines are too long
+4
View File
@@ -22,6 +22,10 @@ dependencies = [
dev = [
"pytest>=8.0",
]
notebook = [
"jupyterlab>=4.2",
"nbclient>=0.10",
]
[project.scripts]
pi05mini-train = "pi05mini.train:main"