hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

ready for review

Browse Source
This commit is contained in:
Alexander Soare
2024-04-16 13:43:58 +01:00
parent 03b08eb74e
commit 9c2f10bd04
4 changed files with 26 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_examples.py
View File

@@ -29,11 +29,12 @@ def test_examples_3_and_2():
with open(path, "r") as file:
file_contents = file.read()
# Do less steps and use CPU.
# Do less steps, use CPU, and don't complicate things with dataloader workers.
file_contents = _find_and_replace(
file_contents,
[
("offline_steps = 5000", "offline_steps = 1"),
("training_steps = 5000", "training_steps = 1"),
("num_workers=4", "num_workers=0"),
('device = torch.device("cuda")', 'device = torch.device("cpu")'),
],
)