admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-20 19:19:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

overwrite the pipeline normalization features with the policy features

Browse Source
This commit is contained in:
Pepijn
2025-09-29 15:22:12 +02:00
parent 57bcf201ad
commit b31fa1e620
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/scripts/lerobot_train.py
+3 -2
View File
@@ -189,9 +189,10 @@ def train(cfg: TrainPipelineConfig):
processor_kwargs["preprocessor_overrides"] = { processor_kwargs["preprocessor_overrides"] = {
"device_processor": {"device": device.type}, "device_processor": {"device": device.type},
"normalizer_processor": {"stats": dataset.meta.stats}, "normalizer_processor": {"stats": dataset.meta.stats},
"normalizer_processor": {"stats": dataset.meta.stats, "features": {**policy.config.input_features, **policy.config.output_features}},
} }
processor_kwargs["postprocessor_overrides"] = { postprocessor_kwargs["postprocessor_overrides"] = {
"unnormalizer_processor": {"stats": dataset.meta.stats}, "unnormalizer_processor": {"stats": dataset.meta.stats, "features": policy.config.output_features},
} }
preprocessor, postprocessor = make_pre_post_processors( preprocessor, postprocessor = make_pre_post_processors(