diff --git a/test/models/test_real_world.py b/test/models/test_real_world.py index 886e3bb4bd..6789bac208 100644 --- a/test/models/test_real_world.py +++ b/test/models/test_real_world.py @@ -86,7 +86,7 @@ class TestRealWorld(unittest.TestCase): loss.backward() optimizer.step() - helper_test("train_cifar", lambda: (Tensor.randn(BS, 3, 32, 32),), train, (1.0/64)*BS, 153) + helper_test("train_cifar", lambda: (Tensor.randn(BS, 3, 32, 32),), train, (1.0/48)*BS, 153) # reset device Tensor.training = old_training