【PyTorch 1.5チュートリアル】レシピ (基本) デバイス間でモデルをセーブ/ロードする

プログラミング
基本カテゴリーからデバイスに渡りモデルをセーブ/ロードする
【PyTorch 1.5チュートリアル】レシピ (基本) デバイス間でモデルをセーブ/ロードする

【PyTorch 1.5 レシピ (基本) デバイス間でモデルをセーブ/ロードする】
今回は「基本」カテゴリーから「デバイスに渡りモデルをセーブ/ロードする」です。このレシピでは、異なるデバイスに渡りニューラルネットワークをセーブしてロードする事例について説明します。

PyTorch は TensorFlow とともにポピュラーな深層学習フレームワークです。PyTorch 1.5 では新たにレシピ集がドキュメントに追加されましたので、これも翻訳しています。

レシピは「基本」「Captum」「データ・カスタマイゼーション」「解釈可能性」「モデル最適化」等のカテゴリーに分かれています。<詳細>

おすすめのコラム