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

Remove b/c workaround

This commit is contained in:
Alexander Soare 2024-03-27 12:03:19 +00:00
parent cb6d1e0871
commit 68d02c80cf
2 changed files with 1 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
README.md
View File

@ -145,7 +145,6 @@ Or you can achieve the same result by executing our script from the command line
```bash ```bash
python lerobot/scripts/eval.py \ python lerobot/scripts/eval.py \
--hub-id lerobot/diffusion_policy_pusht_image \ --hub-id lerobot/diffusion_policy_pusht_image \
--revision v1.0 \
eval_episodes=10 \ eval_episodes=10 \
hydra.run.dir=outputs/eval/example_hub hydra.run.dir=outputs/eval/example_hub
``` ```

7
lerobot/common/datasets/factory.py
View File

@ -81,13 +81,8 @@ def make_offline_buffer(
else: else:
raise ValueError(cfg.env.name) raise ValueError(cfg.env.name)
# TODO(rcadene): backward compatiblity to load pretrained pusht policy
dataset_id = cfg.get("dataset_id")
if dataset_id is None and cfg.env.name == "pusht":
dataset_id = "pusht"
offline_buffer = clsfunc( offline_buffer = clsfunc(
dataset_id=dataset_id, dataset_id=cfg.dataset_id,
sampler=sampler, sampler=sampler,
batch_size=batch_size, batch_size=batch_size,
root=DATA_DIR, root=DATA_DIR,