I’m a beginner in DNN. I’m using Jetson Nano.
I managed to do transfer learning on a ResNet-18 model with my custom dataset for object detection. It seems quite straight forward with Pytorch.
What I’m struggling with is the deployment of my model.
My question is simple: Is it possible to deploy the model that I trained in Pytorch and run object detection inference on it? Or do I absolutely need to export it to some other format such as Caffe?
It it is possible to do it in Pytorch, and if so, is there a sample code in Python?