PyTorch 1.5 レシピ : 基本 : PyTorch の state_dict とは何か (翻訳/解説)
sales-info【PyTorch 1.5 レシピ (基本) state_dict とは何か】
PyTorch は TensorFlow とともにポピュラーな深層学習フレームワークです。PyTorch 1.5 では新たにレシピ集がドキュメントに追加されましたので、これも翻訳しています。
今回は「基本」カテゴリーから「PyTorch の state_dict とは何か」です。このレシピでは state_dict の利用について説明されます。PyTorch では torch.nn.Module モデルの学習可能パラメータはモデルパラメータに含まれます。state_dict は各層をそのパラメータ tensor にマップする、単純な Python 辞書オブジェクトです。
レシピは「基本」「Captum」「データ・カスタマイゼーション」「解釈可能性」「モデル最適化」等のカテゴリーに分かれています。
