PyTorch 1.5 レシピ : 基本 : マルチモデルを一つのファイルでセーブ/ロードする (翻訳/解説)
sales-info【PyTorch 1.5 レシピ (基本) マルチモデルをセーブ/ロードする】
今回のレシピは「マルチモデルを一つのファイルでセーブ/ロードする」です。
GAN, sequence-to-sequence モデルあるいはモデルのアンサンブルのような、マルチ torch.nn.Module から成るモデルをセーブするとき、各モデルの state_dict と対応する optimizer の辞書をセーブしなければなりません。このレシピでは、PyTorch を使用してマルチモデルを一つのファイルにどのようにセーブするかを実演します。
PyTorch は TensorFlow とともにポピュラーな深層学習フレームワークです。PyTorch 1.5 では新たにレシピ集がドキュメントに追加されましたので、これも翻訳しています。
レシピは「基本」「Captum」「データ・カスタマイゼーション」「解釈可能性」「モデル最適化」等のカテゴリーに分かれています。
