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

Do not add model prefix to normalization

Browse Source
This commit is contained in:
Pepijn
2025-09-13 11:25:26 +02:00
parent c5a029a28a
commit 5361346bec
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/pi05_openpi/modeling_pi05openpi.py
+4 -1
View File
@@ -919,7 +919,10 @@ class PI05OpenPIPolicy(PreTrainedPolicy):
remap_count = 0 remap_count = 0
for key, value in fixed_state_dict.items(): for key, value in fixed_state_dict.items():
if not key.startswith("model."): if not key.startswith("model.") and not any(
key.startswith(prefix)
for prefix in ["normalize_inputs.", "normalize_targets.", "unnormalize_outputs."]
):
new_key = f"model.{key}" new_key = f"model.{key}"
remapped_state_dict[new_key] = value remapped_state_dict[new_key] = value
remap_count += 1 remap_count += 1
src/lerobot/policies/pi0_openpi/modeling_pi0openpi.py
+4 -1
View File
@@ -938,7 +938,10 @@ class PI0OpenPIPolicy(PreTrainedPolicy):
remap_count = 0 remap_count = 0
for key, value in fixed_state_dict.items(): for key, value in fixed_state_dict.items():
if not key.startswith("model."): if not key.startswith("model.") and not any(
key.startswith(prefix)
for prefix in ["normalize_inputs.", "normalize_targets.", "unnormalize_outputs."]
):
new_key = f"model.{key}" new_key = f"model.{key}"
remapped_state_dict[new_key] = value remapped_state_dict[new_key] = value
remap_count += 1 remap_count += 1