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

add reward output

Browse Source
This commit is contained in:
Pepijn
2025-11-25 13:44:04 +01:00
parent ca67231892
commit d286ea30d4
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/sarm/configuration_sarm.py
+2 -2
View File
@@ -91,8 +91,8 @@ class SARMConfig(PreTrainedConfig):
"state_features": PolicyFeature(shape=(9, 14), type=FeatureType.STATE) # Example: 7 DOF × 2 arms "state_features": PolicyFeature(shape=(9, 14), type=FeatureType.STATE) # Example: 7 DOF × 2 arms
}) })
output_features: dict = field(default_factory=lambda: { output_features: dict = field(default_factory=lambda: {
"stage": PolicyFeature(shape=(1,), type=FeatureType.OTHER), "stage": PolicyFeature(shape=(1,), type=FeatureType.REWARD),
"progress": PolicyFeature(shape=(1,), type=FeatureType.OTHER) "progress": PolicyFeature(shape=(1,), type=FeatureType.REWARD)
}) })
def __post_init__(self): def __post_init__(self):