add executable pi0.5 reproduction notebook
This commit is contained in:
@@ -5,5 +5,6 @@ __pycache__/
|
||||
*.egg-info/
|
||||
.venv/
|
||||
data/
|
||||
notebook_artifacts/
|
||||
outputs/**/*.pt
|
||||
.DS_Store
|
||||
|
||||
@@ -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
@@ -22,6 +22,10 @@ dependencies = [
|
||||
dev = [
|
||||
"pytest>=8.0",
|
||||
]
|
||||
notebook = [
|
||||
"jupyterlab>=4.2",
|
||||
"nbclient>=0.10",
|
||||
]
|
||||
|
||||
[project.scripts]
|
||||
pi05mini-train = "pi05mini.train:main"
|
||||
|
||||
Reference in New Issue
Block a user