Skip to content

Commit 751bca5

Browse files
committed
added text descriptions
1 parent 7d9dccc commit 751bca5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

advanced_source/super_resolution_with_onnxruntime.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -226,13 +226,13 @@ def to_numpy(tensor):
226226
start = time.time()
227227
torch_out = torch_model(x)
228228
end = time.time()
229-
print(f"Inference used {end - start} seconds")
229+
print(f"Inference of Pytorch model used {end - start} seconds")
230230

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

237237
######################################################################
238238
# Running the model on an image using ONNX Runtime

0 commit comments

Comments
 (0)