Update README.md
Browse files
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 |
```
|