我正在使用 Torch7 库来实现神经网络。大多数情况下,我依赖预训练模型。在 Lua 中,我使用 torch.load 函数加载保存为 torch .t7 文件的模型。我对切换到 PyTorch(http://pytorch.org) 很好奇,我阅读了文档。我找不到有关加载预训练模型的机制的任何信息。我能找到的唯一相关信息是此页面:http://pytorch.org/docs/torch.html
但页面中描述的函数 torch.load 似乎加载了一个用 pickle 保存的文件。如果有人有关于在 PyTorch 中加载 .t7 模型的其他信息,请在此处分享。
最佳答案
正确的函数是load_lua
:
from torch.utils.serialization import load_lua
x = load_lua('x.t7')
关于python - 在 PyTorch 中加载 Torch7 训练模型 (.t7),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41861354/