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