tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

feat(train): include input and output features in processor overrides for normalization (#2088) (#2090)

Browse Source 
Signed-off-by: AdilZouitine <adilzouitinegm@gmail.com>
This commit is contained in:
Adil Zouitine
2025-09-29 17:37:26 +02:00
committed by GitHub
parent 1ad2da403d
commit a0d7627d81
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/lerobot/scripts/lerobot_train.py
View File

@@ -187,10 +187,16 @@ def train(cfg: TrainPipelineConfig):
if cfg.policy.pretrained_path is not None:
processor_kwargs["preprocessor_overrides"] = {
"device_processor": {"device": device.type},
"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"] = {
"unnormalizer_processor": {"stats": dataset.meta.stats},
"unnormalizer_processor": {
"stats": dataset.meta.stats,
"features": {**policy.config.input_features, **policy.config.output_features},
},
}
preprocessor, postprocessor = make_pre_post_processors(