Hi,
I am new in NNVM/TVM, and I encounter accuracy problem when trying to convert pre-trained PyTorch model (ResNet50) to NNVM/TVM.
Problem description:
When I use pre-trained ResNet50 PyTorch model for prediction, the performance is
top 1 accuracy : 0.46
top 5 accuracy : 0.79
However, both accuracies become zero after I convert to TVM and execute.
I also use ONNX-TensorFlow backend to run prediction and the accuracy is
onnx-tf top1 accuracy : 0.5
onnx-tf top5 accuracy : 0.81
What I did
I didn’t find straight way to convert PyTorch to NNVM/TVM, thus I convert PyTorch to ONNX representation first, and use this tutorial to compile ONNX to NNVM and execute using TVM.
How to reproduce
To reproduce my issue, please download my source codes here.
Big Thanks!