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

Quickstart: Use tensors to compute train accuracy (#1662)

Browse Source 
Co-authored-by: Dave Farago <dfarago@innoopract.com>
This commit is contained in:
DavidFarago
2023-08-24 23:09:12 +02:00
committed by GitHub
parent d37d092c14
commit 29adae84eb
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/quickstart.md
View File

@@ -205,11 +205,11 @@ for step in range(1000):
opt.step()
# calculate accuracy
pred = out.argmax(axis=-1).numpy()
pred = out.argmax(axis=-1)
acc = (pred == labels).mean()
if step % 100 == 0:
print(f"Step {step+1} | Loss: {loss.numpy()} | Accuracy: {acc}")
print(f"Step {step+1} | Loss: {loss.numpy()} | Accuracy: {acc.numpy()}")
```
## Evaluation