Reading a trained pytorch model in C++

1

Hi, I have a trained feed forward neural network which uses PReLU and now, I would like to use this model with my c++ code. Is there any way to load pytorch models in C++? I found one wrapper (https://github.com/warmspringwinds/pytorch-cpp) but it does not support PReLU.

1 Like
2

You could try ONNX to convert your model to another C++ framework like Caffe2.

1 Like