github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

oops, timing

Browse Source
This commit is contained in:
George Hotz
2020-11-07 12:15:52 -08:00
parent e4c5fb0219
commit 3ab8fb2189
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/efficientnet.py
View File

@@ -191,8 +191,8 @@ if __name__ == "__main__":
print(np.argmax(out.data), np.max(out.data), lbls[np.argmax(out.data)])
else:
img = Image.open(io.BytesIO(fetch(url)))
out, _ = infer(model, img)
st = time.time()
out, _ = infer(model, img)
print(np.argmax(out.data), np.max(out.data), lbls[np.argmax(out.data)])
print("did inference in %.2f s" % (time.time()-st))
#print("NOT", np.argmin(out.data), np.min(out.data), lbls[np.argmin(out.data)])