Search code examples
machine-learningluacaffeconv-neural-networktorch

Finetune a Caffe model in Torch


I want to finetune a neural net that has been pretrained. However, this model was made in Caffe and I would like to work in Torch.
I have tried loadcaffe, but this does not seem focused on finetuning.
Is there another tool that makes this possible? Or can the Caffe model be converted to a Torch net?


Solution

  • All you need to do is

    1. use loadcaffe to convert your Caffe pretrained network into a Torch version.
      (right after you can save it on disk with torch.save("net.t7", model))
    2. write Torch code to fine-tune it.