python - 从 python 调用 torch7 ( Lua ) 函数？

Question

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

有什么想法吗？

Answer 1

我认为您现在有了更好的解决方案，即 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?