繁体   English   中英

在 Torch 中微调 Caffe 模型

[英]Finetune a Caffe model in Torch

我想微调一个经过预训练的神经网络。 但是,这个模型是在 Caffe 中制作的,我想在 Torch 中工作。
我试过loadcaffe ,但这似乎并不集中在微调上。
有没有其他工具可以使这成为可能? 或者可以将 Caffe 模型转换为 Torch 网?

你需要做的就是

  1. 使用loadcaffe将您的 Caffe 预训练网络转换为 Torch 版本。
    (在您可以使用torch.save("net.t7", model)将其保存在磁盘上torch.save("net.t7", model)
  2. 编写 Torch 代码以对其进行微调。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM