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

update README

Browse Source
This commit is contained in:
Alexander Soare
2024-03-26 16:28:16 +00:00
parent 1ed0110900
commit be4441c7ff
3 changed files with 23 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
tests/test_examples.py
View File

@@ -46,11 +46,25 @@ def test_examples_3_and_2():
with open(path, "r") as file:
file_contents = file.read()
# Do less evals and use CPU.
# Do less evals, use CPU, and use the local model.
file_contents = _find_and_replace(
file_contents,
['"eval_episodes=10"', '"rollout_batch_size=10"', '"device=cuda"'],
['"eval_episodes=1"', '"rollout_batch_size=1"','"device=cpu"'],
[
'"eval_episodes=10"',
'"rollout_batch_size=10"',
'"device=cuda"',
'# folder = Path("outputs/train/example_pusht_diffusion")',
'hub_id = "lerobot/diffusion_policy_pusht_image"',
"folder = Path(snapshot_download(hub_id)",
],
[
'"eval_episodes=1"',
'"rollout_batch_size=1"',
'"device=cpu"',
'folder = Path("outputs/train/example_pusht_diffusion")',
"",
"",
],
)
assert Path(f"outputs/train/example_pusht_diffusion").exists()
assert Path(f"outputs/train/example_pusht_diffusion").exists()