最佳答案
我想你现在有一个更好的解决方案,即lutorpy.
与pytorch不同,你在python中有一个lua引擎,所以在python中导入任何lua模块和代码更灵活,而且它易于使用和灵活.对于pytorch,你只有很少的移植模块,你可以直接在python中使用.
与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?