From fc74001202368287de5c98ff2a5343a9b6d65925 Mon Sep 17 00:00:00 2001 From: AdilZouitine Date: Thu, 10 Jul 2025 17:22:35 +0200 Subject: [PATCH] refactor(train): Update preprocessor initialization to include dataset statistics --- src/lerobot/scripts/train.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/lerobot/scripts/train.py b/src/lerobot/scripts/train.py index 2ab170a1f..5a69b4cad 100644 --- a/src/lerobot/scripts/train.py +++ b/src/lerobot/scripts/train.py @@ -140,7 +140,9 @@ def train(cfg: TrainPipelineConfig): cfg=cfg.policy, ds_meta=dataset.meta, ) - preprocessor, _ = make_processor(cfg.policy, cfg.policy.pretrained_path) + preprocessor, _ = make_processor( + policy_cfg=cfg.policy, pretrained_path=cfg.policy.pretrained_path, dataset_stats=dataset.meta.stats + ) logging.info("Creating optimizer and scheduler") optimizer, lr_scheduler = make_optimizer_and_scheduler(cfg, policy)