2015-11-15 13 views
5

Ich habe ein Programm in Python geschrieben und ich habe Modell ConvNet mit Toch7 trainiert. Ich möchte das Modell aus dem Python-Programm weiterleiten und zurückprovitieren, da ich es schwer und schwer finde, es wieder in lua zu schreiben.Call Fackel (Lua) -Funktion von Python?

Irgendeine Idee bitte?

Antwort

2

Ich denke, Sie haben jetzt eine viel bessere Lösung, die lutorpy ist. Anders als bei pytorch, haben Sie eine Lua-Engine in Python, so ist es flexibler zu importieren alle Lua-Modul und Code in Python, und es ist einfach zu bedienen und flexibel. Für pytorch haben Sie nur sehr wenige portierte Module, die Sie direkt in Python verwenden können.

Mit lutorpy können Sie zwischen numpy und torch tensor einfach und sehr schnell umwandeln.

Für Sie Fall können Sie Ihren Code in Python wie folgt schreiben:

import numpy as np 
import lutorpy as lua 

model = torch.load('PATH TO YOUR MODEL FILE') 

# generate your input data with numpy 
arr = np.random.randn(100) 

# convert your numpy array into torch tensor 
x = torch.fromNumpyArray(arr) 

# apply model forward method with "._" syntax(which is equivalent to ":" in lua) 
y = model._forward(x) 

Ein kurzer Vergleich zwischen verschiedenen Bibliothek: How can I load and use torch deep learning models from python?

+1

pytorch Autor hier. Pytorch enthält eine vollständige Lua-Engine. Es wird entweder luajit (linux) oder lua (mac) laufen. Sie können jetzt Ihre eigenen Lua-Klassen importieren. Siehe zum Beispiel https://github.com/hughperkins/pytorch-residual-networks oder https://github.com/hughperkins/cifar.pytorch –