rizgiak commited on
Commit
b013f8a
·
verified ·
1 Parent(s): 02a7208

Model card: use hf: references and fetch_checkpoints.py

Browse files
Files changed (1) hide show
  1. README.md +9 -7
README.md CHANGED
@@ -51,15 +51,17 @@ Each folder holds `model.safetensors`, `config.json`, `dataset_stats.json`, `tra
51
 
52
  ## Use
53
 
54
- ```python
55
- from huggingface_hub import snapshot_download
56
- root = snapshot_download("rizgiak/physai-study-01-checkpoints", allow_patterns=["act_randomized_look_seed0/*"])
57
- checkpoint = root + "/act_randomized_look_seed0"
 
 
58
  ```
59
 
60
- then pass `checkpoint` as `--checkpoint` to `scripts/eval_policy.py` of the physai framework with the study's
61
- manifest (`research/configs/so101_randomized_pick_place.yaml`) and `--policy lerobot`. For the Diffusion Policy
62
- checkpoints add `--policy-arg n_action_steps=16`.
63
 
64
  Three kinds of seeds appear and they are unrelated:
65
 
 
51
 
52
  ## Use
53
 
54
+ With the physai framework (https://github.com/PhysAI-Robot/physai-robot-starter), `--checkpoint` takes a reference
55
+ to a folder of this repository and downloads it into `checkpoints/` (ignored by git):
56
+
57
+ ```bash
58
+ python scripts/eval_policy.py --manifest research/configs/so101_randomized_pick_place.yaml --policy lerobot --checkpoint hf:rizgiak/physai-study-01-checkpoints/act_randomized_look_seed0@<commit>
59
+ python scripts/fetch_checkpoints.py hf:rizgiak/physai-study-01-checkpoints/act_randomized_look_seed0 # download only, to use offline
60
  ```
61
 
62
+ Name a commit (`@<commit>`) for a result you report; without one `main` is used. For the Diffusion Policy
63
+ checkpoints add `--policy-arg n_action_steps=16`. To use a folder without the framework's downloader, fetch it with
64
+ `huggingface_hub.snapshot_download("rizgiak/physai-study-01-checkpoints", allow_patterns=["<folder>/*"])` and pass its path to `--checkpoint`.
65
 
66
  Three kinds of seeds appear and they are unrelated:
67