github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

chans doesn't need to be in self

Browse Source
This commit is contained in:
George Hotz
2020-10-22 21:19:35 -07:00
parent 77251cc6c3
commit 816f648161
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/test_mnist.py
View File

@@ -25,9 +25,9 @@ class TinyBobNet:
class TinyConvNet: class TinyConvNet:
def __init__(self): def __init__(self):
conv = 7 conv = 7
self.chans = 4 chans = 4
self.c1 = Tensor(layer_init_uniform(self.chans,1,conv,conv)) self.c1 = Tensor(layer_init_uniform(chans,1,conv,conv))
self.l1 = Tensor(layer_init_uniform(((28-conv+1)**2)*self.chans, 128)) self.l1 = Tensor(layer_init_uniform(((28-conv+1)**2)*chans, 128))
self.l2 = Tensor(layer_init_uniform(128, 10)) self.l2 = Tensor(layer_init_uniform(128, 10))
def forward(self, x): def forward(self, x):