模型参数的载入
差别
这里会显示出您选择的修订版和当前版本之间的差别。
| 两侧同时换到之前的修订记录 前一修订版 | |||
|
模型参数的载入 [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
除额外注明的地方外，本维基上的内容按下列许可协议发布： CC Attribution-Share Alike 4.0 International
