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

change expected features

Browse Source
This commit is contained in:
Pepijn
2025-11-25 13:51:01 +01:00
parent d286ea30d4
commit 8d2fb5d298
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/sarm/configuration_sarm.py
+1 -3
View File
@@ -84,10 +84,8 @@ class SARMConfig(PreTrainedConfig):
use_dataset_task: bool = True # Use task descriptions from dataset use_dataset_task: bool = True # Use task descriptions from dataset
use_subtask_annotations: bool = True # Use subtask annotations for stage-aware training if available use_subtask_annotations: bool = True # Use subtask annotations for stage-aware training if available
# Features (required by PreTrainedPolicy) # Video_features and text_features are generated by the processor from raw images/text, we don't declare them as VISUAL/LANGUAGE here to avoid validation errors
input_features: dict = field(default_factory=lambda: { input_features: dict = field(default_factory=lambda: {
"video_features": PolicyFeature(shape=(9, 512), type=FeatureType.VISUAL),
"text_features": PolicyFeature(shape=(384,), type=FeatureType.LANGUAGE),
"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: {