saifahmad123 commited on
Commit
1edf5d2
·
verified ·
1 Parent(s): b0ed3e2

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
+ # EXP3_20_a100
11
+
12
+ openpi π₀.₅ policy checkpoint.
13
+
14
+ - **Train config:** `pi05_Franka_EXP3_20`
15
+ - **Experiment:** `EXP3_20_a100`
16
+ - **Checkpoints in this repo:**
17
+
18
+ - `step_5999/`
19
+
20
+ Each folder is one training step and is self-contained.
21
+ - **Training dataset:** [saifahmad123/EXP3_50](https://huggingface.co/datasets/saifahmad123/EXP3_50)
22
+
23
+ ## Contents
24
+
25
+ | Path | Purpose |
26
+ |---|---|
27
+ | `step_5999/params/` | Model weights. Required to serve the policy. |
28
+ | `step_5999/assets/` | Normalization statistics. Required to serve the policy. |
29
+ | `step_5999/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/EXP3_20_a100", allow_patterns="step_5999/*") + "/step_5999"
39
+ train_config = _config.get_config("pi05_Franka_EXP3_20")
40
+ policy = policy_config.create_trained_policy(train_config, ckpt)
41
+
42
+ action_chunk = policy.infer(observation)["actions"]
43
+ ```