From 3ab8fb2189369564ef0a49e21af49bf6be921e15 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Sat, 7 Nov 2020 12:15:52 -0800 Subject: [PATCH] oops, timing --- examples/efficientnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/efficientnet.py b/examples/efficientnet.py index 5f006b88..1439b2e5 100644 --- a/examples/efficientnet.py +++ b/examples/efficientnet.py @@ -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)])