from torch.utils.serialization import load_lua 出错
时间: 2024-02-28 18:56:25 浏览: 429
Lua 错误处理
如果您在使用 PyTorch 版本为 0.4.0 或更高版本的时候,还是使用 `torch.utils.serialization.load_lua()` 加载 Lua/Torch 模型时出错,那就说明这个函数已经被完全删除了。在这种情况下,您需要使用其他库来加载 Lua/Torch 模型。以下是两个常用的库:
1. `torchfile`:这是一个用来读取和写入 Torch7 文件的 Python 库。您可以使用以下代码来使用 `torchfile` 加载您的模型:
```
import torchfile
model = torchfile.load('model.t7')
```
其中 `model.t7` 是您要加载的模型文件名。
2. `lua`:这是一个用来在 Python 中执行 Lua 代码的库。您可以使用以下代码来使用 `lua` 加载您的模型:
```
from lua import LuaRuntime
lua = LuaRuntime(unpack_returned_tuples=True)
model = lua.require('torch').load('model.t7')
```
其中 `model.t7` 是您要加载的模型文件名。
阅读全文