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

View File

@ -37,7 +37,7 @@ def test_examples_3_and_2():
file_contents,
[
("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")'),
("batch_size=64", "batch_size=1"),
],