Update model card root paths
Browse files
README.md
CHANGED
|
@@ -16,18 +16,17 @@ This repository provides a Pi0.5 checkpoint trained with the VLABench pretrain p
|
|
| 16 |
|
| 17 |
The uploaded checkpoint includes both inference parameters and training/evaluation assets:
|
| 18 |
|
| 19 |
-
- `
|
| 20 |
-
- `
|
| 21 |
-
- `
|
| 22 |
-
- `
|
| 23 |
-
- `torch_199999/`: converted PyTorch checkpoint, including `model.safetensors`, `config.json`, and VLABench assets.
|
| 24 |
|
| 25 |
## Evaluation
|
| 26 |
|
| 27 |
To run this checkpoint, please clone this repo: https://github.com/Shiduo-zh/openpi, and checkout to the branch `main`. Assume that you download this checkpoint and put it in the directory `checkpoints`, to run the policy as server, please run:
|
| 28 |
|
| 29 |
```sh
|
| 30 |
-
bash serve_policy.sh pi05_posttrain_vlabench_primitive_aligned checkpoints/
|
| 31 |
```
|
| 32 |
|
| 33 |
After serving the policy, open another terminal and run:
|
|
|
|
| 16 |
|
| 17 |
The uploaded checkpoint includes both inference parameters and training/evaluation assets:
|
| 18 |
|
| 19 |
+
- `params/`: Orbax parameters for inference/evaluation.
|
| 20 |
+
- `train_state/`: full Orbax training state for resuming training.
|
| 21 |
+
- `assets/`: normalization statistics and checkpoint assets.
|
| 22 |
+
- `_CHECKPOINT_METADATA`: Orbax checkpoint metadata.
|
|
|
|
| 23 |
|
| 24 |
## Evaluation
|
| 25 |
|
| 26 |
To run this checkpoint, please clone this repo: https://github.com/Shiduo-zh/openpi, and checkout to the branch `main`. Assume that you download this checkpoint and put it in the directory `checkpoints`, to run the policy as server, please run:
|
| 27 |
|
| 28 |
```sh
|
| 29 |
+
bash serve_policy.sh pi05_posttrain_vlabench_primitive_aligned checkpoints/pi05-posttrain-vlabench-primitive-aligned/
|
| 30 |
```
|
| 31 |
|
| 32 |
After serving the policy, open another terminal and run:
|