PyTorch 保存模型結構參數及加載模型

保存模型與加載

保存模型分為兩種方式：

保存整個網絡結構和參數

保存整個網絡的參數

# 1.保存并加載整個網絡結構和參數 # 保存模型 torch.save(net, 'net.pth') # 加載模型 net = torch.load('net.pth') # 2.保存并加載整個網絡的參數 # 保存模型 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學習：加載模型和參數

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。