2016-05-10 54 views
1

我想微調一個已經過預訓練的神經網絡。但是,這個模型是在Caffe中製作的,我想在Torch工作。
我已經試過loadcaffe,但是這似乎並不關注微調。
有沒有另一種工具可以使這成爲可能?或者Caffe模型可以轉換成火炬網嗎?在火炬中提供一個Caffe模型

回答

1

所有你需要做的是

  1. 使用loadcaffe來轉換來自Caffe網絡預先訓練成火炬版本。
    (之後您可以將其保存在磁盤上torch.save("net.t7", model)
  2. 寫入火炬代碼以對其進行微調。
+0

正是我需要的。謝謝! – Cassie

+0

@Cassie做了這項工作嗎? –