从python调用torch7(Lua)函数？

发布时间：2020-12-16 22:53:41 所属栏目：Python 来源：网络整理

导读：我有一个程序是用python编写的,我有使用Toch7训练的模型ConvNet.我想从python程序调用forward和backpro到模型,因为我发现很难在lua中再次编写它. 有什么好主意吗？最佳答案我想你现在有一个更好的解决方案,即lutorpy. 与pytorch不同,你在python中有一个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?

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!