File size: 1,236 Bytes
924e104
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
eb7b128
924e104
 
 
 
 
 
 
 
eb7b128
 
 
924e104
 
 
 
 
 
 
 
eb7b128
924e104
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
---
library_name: openpi
tags:
- robotics
- openpi
- pi0
- franka
---

# GRASPNET_FINAL_h100

openpi π₀.₅ policy checkpoint.

- **Train config:** `pi05_Franka_GRASPNET_FINAL`
- **Experiment:** `GRASPNET_FINAL_h100`
- **Checkpoints in this repo:**

- `step_25000/`
- `step_49999/`

Each folder is one training step and is self-contained.
- **Training dataset:** [saifahmad123/GRASPNET_FINAL](https://huggingface.co/datasets/saifahmad123/GRASPNET_FINAL)

## Contents

| Path | Purpose |
|---|---|
| `step_49999/params/` | Model weights. Required to serve the policy. |
| `step_49999/assets/` | Normalization statistics. Required to serve the policy. |
| `step_49999/train_state/` | Optimizer state. Only present if uploaded with `--include-train-state`; needed to resume training. |

## Usage

```python
from huggingface_hub import snapshot_download
from openpi.policies import policy_config
from openpi.training import config as _config

ckpt = snapshot_download("saifahmad123/GRASPNET_FINAL_h100", allow_patterns="step_49999/*") + "/step_49999"
train_config = _config.get_config("pi05_Franka_GRASPNET_FINAL")
policy = policy_config.create_trained_policy(train_config, ckpt)

action_chunk = policy.infer(observation)["actions"]
```