Hi,
One way to go if you have a “classic” network is to export it with onnx and then use tools that can be used for caffe2 or tensorflow models.
The problem with exporting directly from pytorch is that you will usually have a graph with very low level operations which makes it (very) big.