PyTorch torch.load_state_dict 関数
Pytorch torch リファレンスマニュアルtorch.load_state_dictPyTorch で状態辞書をロードするための関数です。辞書オブジェクトからモデルやオプティマイザなどの状態をロードします。
関数定義
torch.load_state_dict(state_dict, strict=True)
使用例
実例
import torch
import torch.nn as nn
# 簡単なモデルを定義する
model = nn.Linear(10, 5)
# モデルの状態辞書を保存する
state_dict = model.state_dict()
torch.save(state_dict, 'model_state.pt')
# 状態辞書をロードする
loaded_state = torch.load('model_state.pt')
model.load_state_dict(loaded_state)
print("State dict loaded successfully!")
import torch.nn as nn
# 簡単なモデルを定義する
model = nn.Linear(10, 5)
# モデルの状態辞書を保存する
state_dict = model.state_dict()
torch.save(state_dict, 'model_state.pt')
# 状態辞書をロードする
loaded_state = torch.load('model_state.pt')
model.load_state_dict(loaded_state)
print("State dict loaded successfully!")
その他の拡張