PyTorchでモデルを保存および読み込む方法は何ですか?
PyTorchでモデルの保存とロードは、以下の手順で行うことができます。
- torch.save()を使用します。
torch.save(model.state_dict(), 'model.pth')
- torch.load()を使用してください。
model = Model()
model.load_state_dict(torch.load('model.pth'))
model.eval()
モデルをロードする際には、保存時と同じモデル構造であることを確認してください。そうでないとロードに失敗する可能性があります。