wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

lint

This commit is contained in:
Ville Kuosmanen 2025-04-13 12:08:52 +02:00
parent 6882dfd012
commit 1dbcf584d6
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lerobot/scripts/train.py
View File

@ -137,10 +137,7 @@ def train(cfg: TrainPipelineConfig):
eval_env = make_env(cfg.env, n_envs=cfg.eval.batch_size, use_async_envs=cfg.eval.use_async_envs)
logging.info("Creating policy")
if isinstance(dataset, MultiLeRobotDataset):
ds_meta = dataset._datasets[0].meta
else:
ds_meta = dataset.meta
ds_meta = dataset._datasets[0].meta if isinstance(dataset, MultiLeRobotDataset) else dataset.meta
policy = make_policy(
cfg=cfg.policy,
ds_meta=ds_meta,