commaai
/
tinygrad
mirror of https://github.com/commaai/tinygrad.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

chans doesn't need to be in self

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
Show Stats 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:
def __init__(self):
conv = 7
self.chans = 4
self.c1 = Tensor(layer_init_uniform(self.chans,1,conv,conv))
self.l1 = Tensor(layer_init_uniform(((28-conv+1)**2)*self.chans, 128))
chans = 4
self.c1 = Tensor(layer_init_uniform(chans,1,conv,conv))
self.l1 = Tensor(layer_init_uniform(((28-conv+1)**2)*chans, 128))
self.l2 = Tensor(layer_init_uniform(128, 10))
def forward(self, x):