So I am new with the tvm/nnvm compiler and I am trying to use some onnx pretrained models to run but I keep getting this error no matter what model I use. I have downloaded them from the official onnx repository but I cant make it work. Anyone can help me?
Traceback (most recent call last):
File "imagenet_classify.py", line 180, in <module>
main()
File "imagenet_classify.py", line 126, in main
net, args.target, shape={"data": img_shape}, params=params)
File "/.local/lib/python3.5/site-packages/nnvm-0.8.0-py3.5.egg/nnvm/compiler/build_module.py", line 250, in build
shape, dtype = _update_shape_dtype(shape, dtype, params)
File "/.local/lib/python3.5/site-packages/nnvm-0.8.0-py3.5.egg/nnvm/compiler/build_module.py", line 136, in _update_shape_dtype
"%s: dtype not expected %s vs %s" % (k, dtype, v.dtype))
ValueError: reshape_attr_tensor421: dtype not expected float32 vs int64