Converted Trained PyTorch Model to Caffe

1

Is there a way to convert a pretrained pytorch model to use in inference in Caffe? I torch.save() the dictionary of the net. How can I get the trained model loaded into Caffe?

2

at this time, there is no such conversion tool.

3

Hi @sauhaardac, have you found any tools to convert trained PyTorch model to Caffe?

4

Nope, I actually ended up converting all our inference code to PyTorch and my entire lab followed suit. No regrets, Pytorch is awesome!

4 Likes