PyTorch 1.5 レシピ : 基本 : 推論のためにモデルをセーブ/ロードする (翻訳/解説)
sales-info【PyTorch 1.5 レシピ (基本) 推論のためにモデルをセーブ/ロードする】
今回は「基本」カテゴリーから「推論のためにモデルをセーブ/ロードする」です。
PyTorch で推論のためにモデルをセーブしてロードするためには 2 つのアプローチがあります。最初のものは state_dict をセーブ/ロードするもので、そして 2 番目はモデル全体をセーブ/ロードします。
PyTorch は TensorFlow とともにポピュラーな深層学習フレームワークです。PyTorch 1.5 では新たにレシピ集がドキュメントに追加されましたので、これも翻訳しています。
レシピは「基本」「Captum」「データ・カスタマイゼーション」「解釈可能性」「モデル最適化」等のカテゴリーに分かれています。
