用户工具

站点工具


模型参数的载入

差别

这里会显示出您选择的修订版和当前版本之间的差别。

到此差别页面的链接

两侧同时换到之前的修订记录 前一修订版
后一修订版
前一修订版
模型参数的载入 [2020/09/02 16:40]
218.104.204.98
模型参数的载入 [2020/09/02 17:07] (当前版本)
218.104.204.98
行 44: 行 44:
  
 ''​torch.load('​modelparameters.pth',​ map_location=lambda storage, loc: storage)''​ ''​torch.load('​modelparameters.pth',​ map_location=lambda storage, loc: storage)''​
 +
 +''​torch.load('​modelparameters.pth',​ map_location='​cpu'​)''​
 +
模型参数的载入.1599036051.txt.gz · 最后更改: 2020/09/02 16:40 由 218.104.204.98