saifahmad123 commited on
Commit
7a8c19b
·
verified ·
1 Parent(s): e4f16a8

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +43 -0
README.md ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: openpi
3
+ tags:
4
+ - robotics
5
+ - openpi
6
+ - pi0
7
+ - franka
8
+ ---
9
+
10
+ # EXP1_5_a100
11
+
12
+ openpi π₀.₅ policy checkpoint.
13
+
14
+ - **Train config:** `pi05_Franka_EXP1_5`
15
+ - **Experiment:** `EXP1_5_a100`
16
+ - **Checkpoints in this repo:**
17
+
18
+ - `step_1999/`
19
+
20
+ Each folder is one training step and is self-contained.
21
+ - **Training dataset:** [saifahmad123/EXP1_5](https://huggingface.co/datasets/saifahmad123/EXP1_5)
22
+
23
+ ## Contents
24
+
25
+ | Path | Purpose |
26
+ |---|---|
27
+ | `step_1999/params/` | Model weights. Required to serve the policy. |
28
+ | `step_1999/assets/` | Normalization statistics. Required to serve the policy. |
29
+ | `step_1999/train_state/` | Optimizer state. Only present if uploaded with `--include-train-state`; needed to resume training. |
30
+
31
+ ## Usage
32
+
33
+ ```python
34
+ 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/EXP1_5_a100", allow_patterns="step_1999/*") + "/step_1999"
39
+ train_config = _config.get_config("pi05_Franka_EXP1_5")
40
+ policy = policy_config.create_trained_policy(train_config, ckpt)
41
+
42
+ action_chunk = policy.infer(observation)["actions"]
43
+ ```