github-actions[bot]
54e6124925
version: sunnypilot v2026.001.000 (dev) date: 2026-05-07T23:07:19 master commit: c28eb958740187620f2282023b8f1997cf90f583
20 lines
751 B
Python
20 lines
751 B
Python
import sys, pickle
|
|
from tinygrad import GlobalCounters
|
|
from tinygrad.helpers import fetch, getenv
|
|
from examples.test_onnx_imagenet import imagenet_dataloader
|
|
|
|
if __name__ == "__main__":
|
|
with open(fetch(sys.argv[1]), "rb") as f:
|
|
run_onnx_jit = pickle.load(f)
|
|
input_name = run_onnx_jit.captured.expected_names[0]
|
|
device = run_onnx_jit.captured.expected_input_info[0][-1]
|
|
print(f"input goes into {input_name=} on {device=}")
|
|
hit = 0
|
|
for i,(img,y) in enumerate(imagenet_dataloader(cnt=getenv("CNT", 100))):
|
|
GlobalCounters.reset()
|
|
p = run_onnx_jit(**{input_name:img.to(device)})
|
|
assert p.shape == (1,1000)
|
|
t = p.to('cpu').argmax().item()
|
|
hit += y==t
|
|
print(f"target: {y:3d} pred: {t:3d} acc: {hit/(i+1)*100:.2f}%")
|