导出
https://pytorch.org/tutorials/advanced/super_resolution_with_onnxruntime.html
不必要的参数:export_params=True, opset_version=10, do_constant_folding=True。
推理
参考 https://zhuanlan.zhihu.com/p/582974246
细节不记录了,主要是以下代码:
sess_options = onnxruntime.SessionOptions()
session = onnxruntime.InferenceSession(export_model_path, sess_options,providers=['CUDAExecutionProvider'])
ort_outputs = session.run(None, ort_inputs)