Hi,
I’m new to TVM and I’m trying to compile a ResNet-50 model and running inference on the PYNQ-Z1 board.
To do so, I use NNVM to compile and save the model as resnet50.json
, resnet50.params
and resnet50.tar
files.
This code shows how I am doing it.
Then, I want to load the previously saved model and upload it to the board. To do so, I use this code.
The problem is that, when I load the graph from the JSON file, TVM raises the following error: Attribute FTVMCompute has not been registered for Operator tvm_op
.
Does anyone know what I’m doing wrong here?
Thanks in advance for your answers.