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

refactor(train): Update preprocessor initialization to include dataset statistics

Browse Source
This commit is contained in:
AdilZouitine
2025-07-10 17:22:35 +02:00
committed by Steven Palma
parent f14ac5d486
commit fc74001202
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/scripts/train.py
+3 -1
View File
@@ -140,7 +140,9 @@ def train(cfg: TrainPipelineConfig):
cfg=cfg.policy, cfg=cfg.policy,
ds_meta=dataset.meta, 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") logging.info("Creating optimizer and scheduler")
optimizer, lr_scheduler = make_optimizer_and_scheduler(cfg, policy) optimizer, lr_scheduler = make_optimizer_and_scheduler(cfg, policy)