ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-31 10:51:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixing action and state dim

Browse Source
This commit is contained in:
Maxime Ellerbach
2026-05-13 08:33:27 +00:00
committed by Maximellerbach
parent 0edb693ee4
commit c6ec8d00e3
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/lerobot/policies/vla_jepa/configuration_vla_jepa.py
View File

@@ -86,6 +86,9 @@ class VLAJEPAConfig(PreTrainedConfig):
raise ValueError("VLAJEPA requires at least one visual input feature.")
if self.action_feature is None:
raise ValueError("VLAJEPA requires an action output feature.")
self.action_dim = self.action_feature.shape[0]
if self.robot_state_feature is not None:
self.state_dim = self.robot_state_feature.shape[0]
def get_optimizer_preset(self) -> AdamWConfig:
return AdamWConfig(