PyTorch 1.5 レシピ : 基本 : 一般的なチェックポイントをセーブ/ロードする (翻訳/解説)
sales-info【PyTorch 1.5 レシピ (基本) チェックポイントをセーブ/ロードする】
今回は「基本」カテゴリーから「一般的なチェックポイントをセーブ/ロードする」です。
推論や訓練を再開するためにチェックポイント・モデルをセーブしてロードすることは有用です。一般的なチェックポイントをセーブするとき、モデルの state_dict だけではなくそれ以上をセーブしなければなりません、optimizer の state_dict もセーブすることも重要です。
PyTorch は TensorFlow とともにポピュラーな深層学習フレームワークです。PyTorch 1.5 では新たにレシピ集がドキュメントに追加されましたので、これも翻訳しています。
レシピは「基本」「Captum」「データ・カスタマイゼーション」「解釈可能性」「モデル最適化」等のカテゴリーに分かれています。
