从python调用torch7(Lua)函数？

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)

不同库之间的简要比较： 如何从 python 加载和使用 torch 深度学习模型？