2017-01-25 2 views
4

Ich benutze Torch7-Bibliothek für die Implementierung von neuronalen Netzen. Meistens bin ich auf vortrainierte Modelle angewiesen. In Lua benutze ich die Funktion fackel.laden, um ein Modell zu laden, das als Fackel-.t7-Datei gespeichert ist. Ich bin neugierig auf den Wechsel zu PyTorch (http://pytorch.org) und ich lese die Dokumente. Ich konnte keine Informationen über die Mechanismen zum Laden eines vortrainierten Modells finden. Die einzige relevante Information, die ich finden konnte, ist diese Seite: http://pytorch.org/docs/torch.htmlLaden von Torch7 trainierten Modellen (.t7) in PyTorch

Aber die Funktion fackel.load auf der Seite beschrieben scheint eine Datei mit Pickle gespeichert zu laden. Wenn jemand weitere Informationen zum Laden von .t7 Modellen in PyTorch hat, teilen Sie es bitte hier mit.

+0

haben einen Blick auf: https://discuss.pytorch.org/t/convert-import-torch-model-to-pytorch/37 – smhx

+0

@smhx Dank viel für die Antwort :) – Arul

Antwort

8

Die korrekte Funktion ist load_lua:

from torch.utils.serialization import load_lua 

x = load_lua('x.t7') 
+0

also, nachdem ich x, wie kann ich es verwenden? – RockTheStar

+0

@RockTheStar: Sie können x.forward() genauso wie andere PyTorch-Modelle verwenden. – Arul

Verwandte Themen