admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 02:06:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

temp: hardcode base model

Browse Source
This commit is contained in:
Pepijn
2025-09-13 14:43:09 +02:00
parent af0676f99e
commit 6db39cad58
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/pi05_openpi/configuration_pi05openpi.py
+1 -1
View File
@@ -67,7 +67,7 @@ class PI05OpenPIConfig(PreTrainedConfig):
device: str | None = None # Device to use for the model (None = auto-detect) device: str | None = None # Device to use for the model (None = auto-detect)
# Pretrained model loading # Pretrained model loading
pretrained_path: str | None = None # Path or repo_id to load pretrained weights from pretrained_path: str | None = "pepijn223/pi0_base_fp32" # Path or repo_id to load pretrained weights from
# Optimizer settings: see openpi `AdamW` and # Optimizer settings: see openpi `AdamW` and
optimizer_lr: float = 2.5e-5 # see openpi `CosineDecaySchedule: peak_lr` optimizer_lr: float = 2.5e-5 # see openpi `CosineDecaySchedule: peak_lr`
src/lerobot/policies/pi0_openpi/configuration_pi0openpi.py
+1 -1
View File
@@ -64,7 +64,7 @@ class PI0OpenPIConfig(PreTrainedConfig):
device: str | None = None # Device to use for the model (None = auto-detect) device: str | None = None # Device to use for the model (None = auto-detect)
# Pretrained model loading # Pretrained model loading
pretrained_path: str | None = None # Path or repo_id to load pretrained weights from pretrained_path: str | None = "pepijn223/pi0_base_fp32" # Path or repo_id to load pretrained weights from
# Optimizer settings: see openpi `AdamW` and # Optimizer settings: see openpi `AdamW` and
optimizer_lr: float = 2.5e-5 # see openpi `CosineDecaySchedule: peak_lr` optimizer_lr: float = 2.5e-5 # see openpi `CosineDecaySchedule: peak_lr`