Question

我有一个用python编写的程序，我使用Toch7训练模型ConvNet。我想从python程序调用forward和backpro到模型，因为我发现很难在lua中再次编写它。

请问好吗？

Answer 1

我认为你现在有了一个更好的解决方案，即lutorpy。与pytorch不同，你在python中有一个lua引擎，所以在python中导入任何lua模块和代码更灵活，而且它易于使用和灵活。对于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?

Answer 2

正如其中一位火炬作者在torch7 maillist上所建议的那样，您可以尝试pytorch。

从python调用torch7（Lua）函数？

2 个答案: