在Pythorch中加载Torch7训练模型（.t7）

Question

我使用Torch7库来实现神经网络。大多数情况下，我依赖于受过训练的模特。在Lua中，我使用torch.load函数加载另存为torch.t7文件的模型。我很想换成Pythorch（http://pytorch.org），我读了这些文件。我找不到任何有关加载预先训练模型的机制的信息。我能找到的唯一相关信息是这个页面：http://pytorch.org/docs/torch.html

但是页面中描述的函数torch.load似乎加载了pickle保存的文件。如果有人有关于在Pythorch中加载.t7模型的附加信息，请在此处共享。

Answer 1

从PyTorch开始，1.0torch.utils.serialization被完全删除。因此，再也没有人可以从Lua Torch进口模型到Pythorch。相反，我建议将PyTorch 0.4.1到pip安装在conda环境中（这样您就可以在此之后删除它），并使用this repo将Lua Torch模型转换为PyTorch模型，而不仅仅是不能用于培训的torch.nn.legacy模型。然后使用PyTorch 1.xx对其执行任何操作。你也可以用PyTorch来训练你的Lua火炬模型：）

Answer 2

正确的函数是load_lua：

from torch.utils.serialization import load_lua

x = load_lua('x.t7')