在PyTorch中加载Torch7训练模型(.t7)

Question

我正在使用Torch7库来实现神经网络。大多数情况下，我依赖于预先训练好的模型。在Lua中，我使用torch.load函数来加载一个保存为torch .t7文件的模型。我对切换到PyTorch( http://pytorch.org)很好奇，所以我阅读了相关文档。我找不到任何关于加载预训练模型的机制的信息。我能找到的唯一相关信息是这个页面：http://pytorch.org/docs/torch.html

但是页面中描述的函数torch.load似乎加载了一个用pickle保存的文件。如果有人有更多关于在PyTorch中加载.t7模型的信息，请在这里分享。

Answer 1

从PyTorch 1.0开始，torch.utils.serialization已完全删除。因此，没有人可以再将模型从Lua Torch导入到PyTorch。相反，我建议在conda环境中通过pip安装Lua Torch 0.4.1 (这样您可以在此之后将其删除)，并使用this repo将您的Lua Torch模型转换为PyTorch模型，而不仅仅是不能用于训练的torch.nn.legacy模型。然后使用PyTorch 1.xx对其执行任何操作。您还可以通过以下方式在PyTorch中训练转换后的Lua Torch模型:)

Answer 2

正确的函数是load_lua

from torch.utils.serialization import load_lua

x = load_lua('x.t7')