小编Aru*_*rul的帖子

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

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

但是页面中描述的函数torch.load似乎加载了一个用pickle保存的文件.如果有人在PyTorch中有关于加载.t7模型的其他信息,请在此处分享.

python torch pre-trained-model pytorch

6
推荐指数
2
解决办法
1万
查看次数

标签 统计

pre-trained-model ×1

python ×1

pytorch ×1

torch ×1