from torch.utils.serialization import load_lua
时间: 2024-02-28 08:56:24 浏览: 171
python torch.utils.data.DataLoader使用方法
5星 · 资源好评率100%
很抱歉,torch.utils.serialization.load_lua() 已经在 PyTorch 0.4.0 版本中被弃用了,可以使用 torchfile 库代替。您可以使用以下代码从 Lua/Torch 中加载模型:
```
import torch
import torchfile
model = torchfile.load('model.t7')
```
其中 `model.t7` 是您要加载的模型文件名。
阅读全文