2016-05-10 17 views
1

Ich möchte ein neuronales Netz verfeinern, das vortrainiert wurde. Dieses Modell wurde jedoch in Caffe hergestellt und ich würde gerne in Torch arbeiten.
Ich habe versucht , aber das scheint nicht auf Feinabstimmung konzentriert.
Gibt es ein anderes Tool, das dies ermöglicht? Oder kann das Caffe-Modell in ein Torchennetz umgewandelt werden?Finetune ein Caffe Modell in Torch

Antwort

1

Alles, was Sie tun müssen, ist

  1. Verwendung loadcaffe Ihr Caffe vortrainierter Netzwerk in eine Taschenlampe Version zu konvertieren.
    (direkt nachdem Sie es auf der Festplatte mit torch.save("net.t7", model) speichern können)
  2. schreiben Sie Torch-Code, um es fein abzustimmen.
+0

Genau was ich brauchte. Vielen Dank! – Cassie

+0

@Cassie hat diese Arbeit gemacht? –