Caffe models

User 2144 | 5/18/2016, 7:01:17 PM

Is there a way to load custom trained caffe models in graphlab ?


User 1207 | 5/18/2016, 9:40:21 PM

Hello @ajinkya,

We can't load caffe models directly, however we can load mxnet models as described in There is a converter from caffe models to mxnet models found in that allows them to be used with mxnet, so you should be set.

Hope that works! -- Hoyt

User 5236 | 6/7/2016, 8:11:40 AM

Can you please show an example of loading a pre-trained MXNet model (such as in into an mxnet.model.FeedForward object?

User 1190 | 6/9/2016, 5:32:48 PM


The API to load model is the same in graphlab.mxnet and opensource mxnet. Please follow the API doc for more details:

Thanks, jay