HaomingSong commited on
Commit
966c146
·
verified ·
1 Parent(s): e0dd4b1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +21 -0
README.md CHANGED
@@ -26,4 +26,25 @@ export LEROBOT_HOME=/PATH/TO/LEROBOT_HOME
26
  uv run scripts/serve_policy.py policy:checkpoint \
27
  --policy.config=pi0_bridge_lora \
28
  --policy.dir=$THE_MODEL_PATH
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
29
  ```
 
26
  uv run scripts/serve_policy.py policy:checkpoint \
27
  --policy.config=pi0_bridge_lora \
28
  --policy.dir=$THE_MODEL_PATH
29
+ ```
30
+
31
+ ```python
32
+ TrainConfig(
33
+ name="pi0_bridge_low_mem_finetune",
34
+ model=pi0.Pi0Config(paligemma_variant="gemma_2b_lora", action_expert_variant="gemma_300m_lora"),
35
+ data=LeRobotBridgeDataConfig(
36
+ repo_id="local/bridge_lerobot",
37
+ base_config=DataConfig(
38
+ local_files_only=True, # Set to True for local-only datasets.
39
+ prompt_from_task=True,
40
+ ),
41
+ ),
42
+ weight_loader=weight_loaders.CheckpointWeightLoader("s3://openpi-assets/checkpoints/pi0_base/params"),
43
+ num_train_steps=30_000,
44
+ freeze_filter=pi0.Pi0Config(
45
+ paligemma_variant="gemma_2b_lora", action_expert_variant="gemma_300m_lora"
46
+ ).get_freeze_filter(),
47
+ ema_decay=None,
48
+ num_workers=8,
49
+ ),
50
  ```