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

oops, timing

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
Show Stats 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)])