PyTorch 1.5 レシピ : 基本 : PyTorch でニューラルネットワークを定義する (翻訳/解説)
sales-info【PyTorch 1.5 レシピ (基本) ニューラルネットを定義する】
PyTorch は TensorFlow とともにポピュラーな深層学習フレームワークです。PyTorch 1.5 では新たにレシピ集がドキュメントに追加されましたので、これも翻訳しています。
今回は「基本」カテゴリーから「PyTorch でニューラルネットワークを定義する」です。このレシピでは、MNIST データセット を対象とするニューラルネットワークを定義するために torch.nn を利用します。ニューラルネットワークを作成して訓練することを助けるために PyTorch はエレガントに設計された、torch.nn を含む、モジュールとクラスを提供します。
レシピは「基本」「Captum」「データ・カスタマイゼーション」「解釈可能性」「モデル最適化」等のカテゴリーに分かれています。
