We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7d9dccc commit 751bca5Copy full SHA for 751bca5
advanced_source/super_resolution_with_onnxruntime.py
@@ -226,13 +226,13 @@ def to_numpy(tensor):
226
start = time.time()
227
torch_out = torch_model(x)
228
end = time.time()
229
-print(f"Inference used {end - start} seconds")
+print(f"Inference of Pytorch model used {end - start} seconds")
230
231
232
ort_inputs = {ort_session.get_inputs()[0].name: to_numpy(x)}
233
ort_outs = ort_session.run(None, ort_inputs)
234
235
+print(f"Inference of ONNX model used {end - start} seconds")
236
237
######################################################################
238
# Running the model on an image using ONNX Runtime
0 commit comments