yypeng666 commited on
Commit
95cc108
·
verified ·
1 Parent(s): e97897e

Add checkpoint manifest

Browse files
Files changed (2) hide show
  1. README.md +42 -0
  2. checkpoint_manifest.json +33 -0
README.md ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: openpi
3
+ tags:
4
+ - openpi
5
+ - pi0.5
6
+ - lora
7
+ - lerobot
8
+ - robotics
9
+ ---
10
+
11
+ # OpenPI pi0.5 LoRA on sim-table
12
+
13
+ This repository stores OpenPI checkpoints for the `pi05_clutter_libero_lora`
14
+ training run on `IDEAS-Lab-Northwestern/sim-table`.
15
+
16
+ The checkpoint layout mirrors OpenPI's local checkpoint tree so later training
17
+ checkpoints can be appended without reshaping the repo:
18
+
19
+ ```text
20
+ checkpoints/
21
+ pi05_clutter_libero_lora/
22
+ sim_table_lora/
23
+ 3000/
24
+ _CHECKPOINT_METADATA
25
+ params/
26
+ train_state/
27
+ assets/
28
+ ```
29
+
30
+ To add another checkpoint later, upload the new numeric step directory to the
31
+ same run path. For example:
32
+
33
+ ```bash
34
+ UV_CACHE_DIR=/tmp/uv-cache-sim-table \
35
+ uv run huggingface-cli upload IDEAS-Lab-Northwestern/pi05-sim-table-lora \
36
+ checkpoints/pi05_clutter_libero_lora/sim_table_lora/4000 \
37
+ checkpoints/pi05_clutter_libero_lora/sim_table_lora/4000 \
38
+ --repo-type model \
39
+ --commit-message "Add checkpoint step 4000"
40
+ ```
41
+
42
+ The associated LeRobot dataset is `IDEAS-Lab-Northwestern/sim-table`.
checkpoint_manifest.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "repo_id": "IDEAS-Lab-Northwestern/pi05-sim-table-lora",
3
+ "repo_type": "model",
4
+ "created_utc": "2026-04-29T19:41:54Z",
5
+ "source_git_commit": "650c5b0283a49c42784fb5055a0507da2c6d347d",
6
+ "train_config_name": "pi05_clutter_libero_lora",
7
+ "experiment_name": "sim_table_lora",
8
+ "dataset_repo_id": "IDEAS-Lab-Northwestern/sim-table",
9
+ "checkpoint_base_path": "checkpoints/pi05_clutter_libero_lora/sim_table_lora",
10
+ "checkpoints": [
11
+ {
12
+ "step": 3000,
13
+ "path": "checkpoints/pi05_clutter_libero_lora/sim_table_lora/3000",
14
+ "local_source": "checkpoints/pi05_clutter_libero_lora/sim_table_lora/3000",
15
+ "size": "9.0G",
16
+ "files": 27
17
+ }
18
+ ],
19
+ "model": {
20
+ "family": "pi0.5",
21
+ "action_dim": 32,
22
+ "action_horizon": 16,
23
+ "paligemma_variant": "gemma_2b_lora",
24
+ "action_expert_variant": "gemma_300m_lora",
25
+ "discrete_state_input": false
26
+ },
27
+ "training": {
28
+ "num_train_steps": 20000,
29
+ "batch_size": 4,
30
+ "ema_decay": null,
31
+ "base_checkpoint": "gs://openpi-assets/checkpoints/pi05_base/params"
32
+ }
33
+ }