saifahmad123 commited on
Commit
eb7b128
·
verified ·
1 Parent(s): ecc0570

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +5 -4
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
- | `step_25000/params/` | Model weights. Required to serve the policy. |
28
- | `step_25000/assets/` | Normalization statistics. Required to serve the policy. |
29
- | `step_25000/train_state/` | Optimizer state. Only present if uploaded with `--include-train-state`; needed to resume training. |
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="step_25000/*") + "/step_25000"
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