在Torch中微调Caffe模型

我想微调一个已经预训练过的神经网络。但是,这个模型是用Caffe建立的,我想在Torch中工作。

我已经尝试使用loadcaffe,但这似乎不是针对微调的。

是否有另一种工具可以实现这一点?或者可以将Caffe模型转换为Torch网络吗?

点赞
用户1688185
用户1688185

你所需要做的是:

  1. 使用 loadcaffe 将你的 Caffe预训练网络转换成 Torch 版本。

    (转换完成后,你可以使用 torch.save("net.t7", model) 将其保存到磁盘上)

  2. 编写 Torch 代码对其进行微调。

2016-05-10 13:52:36