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

more readme

Browse Source
This commit is contained in:
George Hotz
2020-10-18 14:38:20 -07:00
parent cc9054e3ec
commit 4019c38942
2 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/mnist.py
View File

@@ -41,14 +41,14 @@ for i in (t := trange(1000)):
y = Tensor(y)
# network
outs = model.forward(x)
out = model.forward(x)
# NLL loss function
loss = outs.mul(y).mean()
loss = out.mul(y).mean()
loss.backward()
optim.step()
cat = np.argmax(outs.data, axis=1)
cat = np.argmax(out.data, axis=1)
accuracy = (cat == Y).mean()
# printing