如何从python加载和使用火炬深度学习模型?

ero*_*gol 2 lua python-2.7 deep-learning torch

我使用python作为我的真实应用程序,并喜欢在python中使用火炬模型.有没有可用的界面或简单的方法呢?

Wei*_*Wei 5

有几个库可以让你在python中与火炬互动,但我会推荐lutorpy,这里有一些细节:

  • lutorpy:最有希望用火炬深度学习的,它基于lupa,更容易使用(更pythonic).它在python中保留了lua引擎,所有lua变量和函数都自动转换为python对象.因此,您可以使用"require"功能导入任何火炬模块.您可以轻松地将几乎任何lua/torch代码转换为python,只需极少的更改.它还为numpy数组和割炬张量提供转换函数,转换是即时的,因为它们共享底层内存.

  • lunatic-python:在python中使用lua的最小库,但功能有限

  • lupa:一个在python中使用lua的库,有很多有用的功能,但它一般用于lua,而不是火炬,例如你不能轻易地将火炬张量转换为numpy数组.

  • pytorch:火炬包装,代码从火炬中取出并编译成python模块,所以你可以使用火炬pythonicly,但你可以使用的模块是非常有限的(到目前为止只支持nn).

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