Lynn777 commited on
Commit
f46c3f4
·
verified ·
1 Parent(s): afb2196

Update model card root paths

Browse files
Files changed (1) hide show
  1. README.md +5 -6
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
- - `jax_pi05_base_aligned_sg/199999/params/`: Orbax parameters for inference/evaluation.
20
- - `jax_pi05_base_aligned_sg/199999/train_state/`: full Orbax training state for resuming training.
21
- - `jax_pi05_base_aligned_sg/199999/assets/`: normalization statistics and checkpoint assets.
22
- - `jax_pi05_base_aligned_sg/199999/_CHECKPOINT_METADATA`: Orbax checkpoint metadata.
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/pi05_posttrain_vlabench_primitive_aligned/jax_pi05_base_aligned_sg/199999/
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: