您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

从python调用torch7(Lua)函数?

5b51 2022/1/14 8:22:06 python 字数 1922 阅读 541 来源 www.jb51.cc/python

我有一个程序是用python编写的,我有使用Toch7训练的模型ConvNet.我想从python程序调用forward和backpro到模型,因为我发现很难在lua中再次编写它.有什么好主意吗?最佳答案我想你现在有一个更好的解决方案,即lutorpy.与pytorch不同,你在python中有一个lua引擎,所以在python中导入任何lua模块和代码更灵

概述

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

有什么好主意吗?

使用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调用torch7(Lua)函数?全部内容,希望文章能够帮你解决从python调用torch7(Lua)函数?所遇到的程序开发问题。


如果您也喜欢它,动动您的小指点个赞吧

除非注明,文章均由 laddyq.com 整理发布,欢迎转载。

转载请注明:
链接:http://laddyq.com
来源:laddyq.com
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。


联系我
置顶