How to load caffe models in pytorch

[kirk86]

Hi everyone, with torch we had loadcaffe which was extremely useful. Is there any way to achieve the same functionality in pytorch? Simply how can we load caffe models in pytorch?

Thanks a lot!

[apaszke]

One way would be to use loadcaffe to load weights in Lua, save them, and use torch.utils.serialization.load_lua to load them in PyTorch. Unfortunately we don't have a ready solution for that yet.

[kirk86]

Thanks Adam,
hope there is some plan for future solution?

[apaszke]

Not anytime soon I'm afraid. There are a lot of high priority tasks.

[apaszke]

But it should be quite simple for someone to add that. You just need to read protobufs and translate graphs into modules.

[kirk86]

gotcha ya. Thanks a lot for the suggestions.

[xiaoxing_zeng]

@apaszke,you means that I can load caffe models in Pytorch using torch?

[apaszke]

Yes it should be possible. But these will only be torch.legacy.nn models, not torch.nn networks.