我想微调经过预训练的神经网络。但是,此模型是在Caffe中制作的,我想在Torch中工作。
我已经尝试过loadcaffe
,但是这似乎并不专注于微调。
有没有其他工具可以实现这一目标?还是可以将Caffe模型转换为Torch网络?
最佳答案
您需要做的就是
使用loadcaffe将经过Caffe训练的网络转换为Torch版本。
(在您可以使用torch.save("net.t7", model)
将其保存在磁盘上之后)
编写Torch代码以对其进行微调。
关于machine-learning - 在Torch中微调Caffe模型,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/37140347/