这里会显示出您选择的修订版和当前版本之间的差别。
两侧同时换到之前的修订记录 前一修订版 后一修订版 | 前一修订版 | ||
模型参数的载入 [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')'' | ||
+ |