PyTorchモデル構造パラメータおよびロードモデルの保存

1704 ワード

PyTorch python 深度学習人工知能ネットワーク

PyTorchモデル構造パラメータおよびロードモデルの保存
モデルとロードの保存
モデルを保存するには、次の2つの方法があります.

は、ネットワーク構造全体およびパラメータ

を保持する.

ネットワーク全体のパラメータ

を保存する

# 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学習:モデルとパラメータのロード