Skip to content

Commit e992e4d

Browse files
committed
taken to_numpy() out of timing script
1 parent 914d168 commit e992e4d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

advanced_source/super_resolution_with_onnxruntime.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -228,8 +228,8 @@ def to_numpy(tensor):
228228
end = time.time()
229229
print(f"Inference of Pytorch model used {end - start} seconds")
230230

231-
start = time.time()
232231
ort_inputs = {ort_session.get_inputs()[0].name: to_numpy(x)}
232+
start = time.time()
233233
ort_outs = ort_session.run(None, ort_inputs)
234234
end = time.time()
235235
print(f"Inference of ONNX model used {end - start} seconds")

0 commit comments

Comments
 (0)