如何从Python中加载和使用Torch深度学习模型?

我在我的实际应用程序中使用Python,想要在Python中利用Torch模型。是否有任何可用的接口或某种方法可以实现?

点赞
用户6262499
用户6262499

有几个库可以让你在Python中与Torch交互,但我会推荐 lutorpy,以下是一些细节:

  • lutorpy:这是用于使用Torch进行深度学习的最有前途的库,它基于lupa,更易于使用(更Pythonic)。它在Python中保留了Lua引擎,所有Lua变量和函数都会自动转换为Python对象。因此,您可以通过使用“require”函数导入任何Torch模块。您可以轻松地将几乎任何Lua / Torch代码转换为Python,只需进行极小的更改即可。它还提供了用于numpy数组和torch张量的转换函数,由于它们共享基础内存,因此转换是即时的。

  • lunatic-python:一个最小化的用于在Python中使用Lua的库,但其功能有限。

  • lupa: 一个用于在Python中使用Lua的库,具有许多有用的功能,但它是针对一般的Lua,而不是Torch,例如您无法轻松地将Torch张量转换成numpy数组。

  • pytorch: 一个Torch的封装,代码是从Torch中取出并编译成Python模块的,因此您可以使用Pythonic的Torch,但可以使用的模块非常有限(到目前为止只支持nn)。

这里是一个例子:https://stackoverflow.com/a/36896347/6262499

2016-04-27 16:44:54