mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages 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
commit 29adae84eb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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