rvsagar9567 commited on
Commit
c00a2f1
·
verified ·
1 Parent(s): 394a158

Document step_5000 + step_7500 checkpoints

Browse files
Files changed (1) hide show
  1. README.md +8 -7
README.md CHANGED
@@ -78,14 +78,15 @@ If `pi05_piper_stacking` is not yet in your `src/openpi/training/config.py`, add
78
  Fine-tuned from Physical Intelligence's `pi05_base` checkpoint via openpi; the weights inherit the
79
  upstream openpi / base-model license. Dataset: axiboai/piper_stacking.
80
 
81
-
82
  ## Checkpoints in this repo
83
 
84
- Two checkpoints from the **same run** are provided so you can compare for overfitting:
85
 
86
- | Path | Step | Train loss | Load with |
87
- |---|---|---|---|
88
- | `/` (root) | 10000 (final) | 0.0012 | `create_trained_policy(cfg, ckpt_dir)` |
89
- | `step_2500/` | 2500 | 0.0048 | `create_trained_policy(cfg, ckpt_dir / "step_2500")` |
 
 
90
 
91
- Training loss fell monotonically (2500: 0.0048 5000: 0.0031 7500: 0.0021 → final: 0.0012). Train loss alone does not reveal overfitting compare **real-rollout success rate** of `step_2500/` vs the final to check whether the longer run over-memorized the 60 demos.
 
78
  Fine-tuned from Physical Intelligence's `pi05_base` checkpoint via openpi; the weights inherit the
79
  upstream openpi / base-model license. Dataset: axiboai/piper_stacking.
80
 
 
81
  ## Checkpoints in this repo
82
 
83
+ Checkpoints from the **same run** (for overfitting comparison):
84
 
85
+ | Path | Step | Train loss |
86
+ |---|---|---|
87
+ | `/` (root) | 10,000 (final) | 0.0012 |
88
+ | `step_7500/` | 7,500 | 0.0021 |
89
+ | `step_5000/` | 5,000 | 0.0031 |
90
+ | `step_2500/` | 2,500 | 0.0048 |
91
 
92
+ Load any with `create_trained_policy(cfg, ckpt_dir / "step_XXXX")` (root = final). Training loss fell monotonically (2,500 final); compare **real-rollout success** to tell whether the longer run overfit the 60 demos.