PyTorch 保存模型結(jié)構(gòu)參數(shù)及加載模型

保存模型與加載

保存模型分為兩種方式：

保存整個網(wǎng)絡(luò)結(jié)構(gòu)和參數(shù)

保存整個網(wǎng)絡(luò)的參數(shù)

# 1.保存并加載整個網(wǎng)絡(luò)結(jié)構(gòu)和參數(shù) # 保存模型 torch.save(net, 'net.pth') # 加載模型 net = torch.load('net.pth') # 2.保存并加載整個網(wǎng)絡(luò)的參數(shù) # 保存模型 torch.save(net.state_dict(), 'net.pth') # 加載模型 net.load_state_dict(torch.load('net.pth'))

Ref

https://pytorch.org/tutorials/beginner/saving_loading_models.html

PyTorch學(xué)習(xí)：加載模型和參數(shù)

總結(jié)

以上是生活随笔為你收集整理的PyTorch 保存模型结构参数及加载模型的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。