Skip to content

Commit 7c2ea85

Browse files
authored
Raise memory limit for CIFAR test (tinygrad#1499)
1 parent 293a102 commit 7c2ea85

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/models/test_real_world.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ def train(X):
8686
loss.backward()
8787
optimizer.step()
8888

89-
helper_test("train_cifar", lambda: (Tensor.randn(BS, 3, 32, 32),), train, (1.0/64)*BS, 153)
89+
helper_test("train_cifar", lambda: (Tensor.randn(BS, 3, 32, 32),), train, (1.0/48)*BS, 153)
9090

9191
# reset device
9292
Tensor.training = old_training

0 commit comments

Comments
 (0)