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

try

This commit is contained in:
Cadene 2024-05-02 21:50:04 +00:00
parent dab8539f44
commit 9e47da81ff
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_examples.py
View File

@ -37,7 +37,7 @@ def test_examples_3_and_2():
file_contents, file_contents,
[ [
("training_steps = 5000", "training_steps = 1"), ("training_steps = 5000", "training_steps = 1"),
("num_workers=4", "num_workers=0"), # ("num_workers=4", "num_workers=0"),
('device = torch.device("cuda")', 'device = torch.device("cpu")'), ('device = torch.device("cuda")', 'device = torch.device("cpu")'),
("batch_size=64", "batch_size=1"), ("batch_size=64", "batch_size=1"),
], ],