测试 ONNX relay 模型

测试 ONNX relay 模型#

import set_env

import onnx
from tvm import relay

onnx_model = onnx.load(path)
mod, params = relay.frontend.from_onnx(onnx_model)#shape_dict, freeze_params=True)

---------------------------------------------------------------------------
NameError                                 Traceback (most recent call last)
Cell In[2], line 4
      1 import onnx
      2 from tvm import relay
----> 4 onnx_model = onnx.load(path)
      5 mod, params = relay.frontend.from_onnx(onnx_model)

NameError: name 'path' is not defined

mod.show()