2015-11-15 13 views
5

私はPythonで書かれたプログラムを持っており、Tov7を使って訓練されたモデルConvNetを持っています。私は、ルアで再び書くのが難しくて難しいと思っているので、pythonプログラムからforwardとbackproをモデルに呼びたいと思います。pythonからtorch7(Lua)関数を呼び出しますか?

どうぞよろしくお願いします。

答えて

2

私はあなたに今より良い解決策があると思います。これはlutorpyです。 pytorchとは異なり、Pythonでluaエンジンを使用していますので、任意のluaモジュールとPythonでコードをインポートする方が柔軟性があり、使いやすく柔軟です。 Pytorchの場合は、Pythonで直接使用できる移植されたモジュールはほとんどありません。

lutorpyでは、numpyとtorchテンソルの間の変換を簡単かつ高速に行うことができます。

あなたの場合のために、あなたはこのようにPythonでコードを書くことができます。

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) 

異なるライブラリ間の簡単な比較:ここ How can I load and use torch deep learning models from python?

+1

pytorchの作者。 pytorchには完全なluaエンジンが含まれています。 luajit(linux)またはlua(mac)を実行します。独自のluaクラスをインポートすることができます。たとえば、https://github.com/hughperkins/pytorch-residual-networksまたはhttps://github.com/hughperkins/cifar.pytorchを参照してください。 –

1

Torchの著者の1人であるtorch7 maillistが示唆したように、pytorchを試すことができます。

関連する問題