Hi everyone,
So I’m trying to compile my ONNX Pytorch model, the first one works but the second model gives “KeyError: ‘input.1’” as error, even though that’s the input together with “1”.
Here is the piece of code:
shape_dict_matrix = { ‘input.1’: (1,256,64,64),‘1’: (1,256,64,83) }
mod_matrix, params_matrix = relay.frontend.from_onnx(matrix_model, shape_dict_matrix)
with relay.build_config(opt_level=0):
matrix_executor = relay.build_module.create_executor('graph', mod_matrix, tvm.gpu(0), target)
dtype = ‘float32’
matrix_output = matrix_executor.evaluate()(tvm.nd.array(vgg_output.astype(dtype)), **params_matrix).asnumpy()
And here’s the full error:
File "tvm_from_onnx.py", line 99, in <module>
matrix_output = matrix_executor.evaluate()(tvm.nd.array(vgg_output.astype(dtype)), **params_matrix).asnumpy()
File "C:\Users\Alexa\AppData\Roaming\Python\Python37\site-packages\tvm-0.6.0-py3.7-win-amd64.egg\tvm\relay\build_module.py", line 333, in _graph_wrapper
args = self._convert_args(self.mod["main"], args, kwargs)
File "C:\Users\Alexa\AppData\Roaming\Python\Python37\site-packages\tvm-0.6.0-py3.7-win-amd64.egg\tvm\relay\backend\interpreter.py", line 194, in _convert_args
cargs.append(kwargs[name])
KeyError: 'input.1'
I did use this pull request https://github.com/apache/incubator-tvm/pull/4536#issuecomment-570700571 to make the Resize op work.