Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -16,6 +16,7 @@ openpi π₀.₅ policy checkpoint.
|
|
| 16 |
- **Checkpoints in this repo:**
|
| 17 |
|
| 18 |
- `step_25000/`
|
|
|
|
| 19 |
|
| 20 |
Each folder is one training step and is self-contained.
|
| 21 |
- **Training dataset:** [saifahmad123/GRASPNET_FINAL](https://huggingface.co/datasets/saifahmad123/GRASPNET_FINAL)
|
|
@@ -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_h100", allow_patterns="
|
| 39 |
train_config = _config.get_config("pi05_Franka_GRASPNET_FINAL")
|
| 40 |
policy = policy_config.create_trained_policy(train_config, ckpt)
|
| 41 |
|
|
|
|
| 16 |
- **Checkpoints in this repo:**
|
| 17 |
|
| 18 |
- `step_25000/`
|
| 19 |
+
- `step_49999/`
|
| 20 |
|
| 21 |
Each folder is one training step and is self-contained.
|
| 22 |
- **Training dataset:** [saifahmad123/GRASPNET_FINAL](https://huggingface.co/datasets/saifahmad123/GRASPNET_FINAL)
|
|
|
|
| 25 |
|
| 26 |
| Path | Purpose |
|
| 27 |
|---|---|
|
| 28 |
+
| `step_49999/params/` | Model weights. Required to serve the policy. |
|
| 29 |
+
| `step_49999/assets/` | Normalization statistics. Required to serve the policy. |
|
| 30 |
+
| `step_49999/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_h100", allow_patterns="step_49999/*") + "/step_49999"
|
| 40 |
train_config = _config.get_config("pi05_Franka_GRASPNET_FINAL")
|
| 41 |
policy = policy_config.create_trained_policy(train_config, ckpt)
|
| 42 |
|