CODEWITHSUNDEEP
machine-learningluacaffeconv-neural-networktorch
Cassie
Cassie

Reputation: 372

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?

Upvotes: 1

Views: 534

Answers (1)

deltheil
deltheil

Reputation: 16121

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.

Upvotes: 1

Related Questions