从python调用torch7(Lua)函数?

前端之家收集整理的这篇文章主要介绍了从python调用torch7(Lua)函数?前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

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

有什么好主意吗?

最佳答案
我想你现在有一个更好的解决方案,即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?

猜你在找的Python相关文章