saifahmad123 commited on
Commit
025a3c3
·
verified ·
1 Parent(s): 0b668c0

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +5 -4
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
- | `step_99999/params/` | Model weights. Required to serve the policy. |
28
- | `step_99999/assets/` | Normalization statistics. Required to serve the policy. |
29
- | `step_99999/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_moreData_h100", allow_patterns="step_99999/*") + "/step_99999"
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