nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity

cleanups

Browse Source
This commit is contained in:
geohot
2025-08-11 08:57:30 -07:00
parent 04fa825a26
commit 2feeb8c8a6
2 changed files with 10 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_tiny.py
+6 -6
View File
@@ -91,12 +91,12 @@ class TestTiny(unittest.TestCase):
@unittest.skipIf(IMAGE>0 or (CI and Device.DEFAULT == "DSP"), "failing because of make things that can't be images not images")
def test_mnist(self):
layers = [
nn.Conv2d(1, 32, 5), Tensor.relu,
nn.Conv2d(32, 32, 5), Tensor.relu,
nn.BatchNorm(32), Tensor.max_pool2d,
nn.Conv2d(32, 64, 3), Tensor.relu,
nn.Conv2d(64, 64, 3), Tensor.relu,
nn.BatchNorm(64), Tensor.max_pool2d,
nn.Conv2d(1, 32, 5), Tensor.relu, Tensor.contiguous,
nn.Conv2d(32, 32, 5), Tensor.relu, Tensor.contiguous,
nn.BatchNorm(32), Tensor.max_pool2d, Tensor.contiguous,
nn.Conv2d(32, 64, 3), Tensor.relu, Tensor.contiguous,
nn.Conv2d(64, 64, 3), Tensor.relu, Tensor.contiguous,
nn.BatchNorm(64), Tensor.max_pool2d, Tensor.contiguous,
lambda x: x.flatten(1), nn.Linear(576, 10)]
# replace random weights with ones