Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -15,6 +15,7 @@ openpi π₀.₅ policy checkpoint.
|
|
| 15 |
- **Experiment:** `GRASPNET_FINAL_moreData_h100`
|
| 16 |
- **Checkpoints in this repo:**
|
| 17 |
|
|
|
|
| 18 |
- `step_99999/`
|
| 19 |
|
| 20 |
Each folder is one training step and is self-contained.
|
|
@@ -24,9 +25,9 @@ Each folder is one training step and is self-contained.
|
|
| 24 |
|
| 25 |
| Path | Purpose |
|
| 26 |
|---|---|
|
| 27 |
-
| `
|
| 28 |
-
| `
|
| 29 |
-
| `
|
| 30 |
|
| 31 |
## Usage
|
| 32 |
|
|
@@ -35,7 +36,7 @@ from huggingface_hub import snapshot_download
|
|
| 35 |
from openpi.policies import policy_config
|
| 36 |
from openpi.training import config as _config
|
| 37 |
|
| 38 |
-
ckpt = snapshot_download("saifahmad123/GRASPNET_FINAL_moreData_h100", allow_patterns="
|
| 39 |
train_config = _config.get_config("pi05_Franka_GRASPNET_FINAL_moreData")
|
| 40 |
policy = policy_config.create_trained_policy(train_config, ckpt)
|
| 41 |
|
|
|
|
| 15 |
- **Experiment:** `GRASPNET_FINAL_moreData_h100`
|
| 16 |
- **Checkpoints in this repo:**
|
| 17 |
|
| 18 |
+
- `step_50000/`
|
| 19 |
- `step_99999/`
|
| 20 |
|
| 21 |
Each folder is one training step and is self-contained.
|
|
|
|
| 25 |
|
| 26 |
| Path | Purpose |
|
| 27 |
|---|---|
|
| 28 |
+
| `step_50000/params/` | Model weights. Required to serve the policy. |
|
| 29 |
+
| `step_50000/assets/` | Normalization statistics. Required to serve the policy. |
|
| 30 |
+
| `step_50000/train_state/` | Optimizer state. Only present if uploaded with `--include-train-state`; needed to resume training. |
|
| 31 |
|
| 32 |
## Usage
|
| 33 |
|
|
|
|
| 36 |
from openpi.policies import policy_config
|
| 37 |
from openpi.training import config as _config
|
| 38 |
|
| 39 |
+
ckpt = snapshot_download("saifahmad123/GRASPNET_FINAL_moreData_h100", allow_patterns="step_50000/*") + "/step_50000"
|
| 40 |
train_config = _config.get_config("pi05_Franka_GRASPNET_FINAL_moreData")
|
| 41 |
policy = policy_config.create_trained_policy(train_config, ckpt)
|
| 42 |
|