在 Torch 中微调 Caffe 模型
Finetune a Caffe model in Torch
我想微调一个已经预训练的神经网络。不过这个模型是用Caffe做的,我想用Torch做。
我已经试过了loadcaffe
,但这似乎并不专注于微调。
是否有另一种工具可以使这成为可能?或者 Caffe 模型可以转换为 Torch 网络吗?
您只需
- 使用 loadcaffe 将您的 Caffe 预训练网络转换为 Torch 版本。
(在您可以使用 torch.save("net.t7", model)
将其保存到磁盘之后)
- 编写 Torch 代码对其进行微调。
我想微调一个已经预训练的神经网络。不过这个模型是用Caffe做的,我想用Torch做。
我已经试过了loadcaffe
,但这似乎并不专注于微调。
是否有另一种工具可以使这成为可能?或者 Caffe 模型可以转换为 Torch 网络吗?
您只需
- 使用 loadcaffe 将您的 Caffe 预训练网络转换为 Torch 版本。
(在您可以使用torch.save("net.t7", model)
将其保存到磁盘之后) - 编写 Torch 代码对其进行微调。