今更PyTorchマスターへの道 (7) - 実際にネットワークを組んで学習&テスト
今日もPython Engineerのチュートリアルをこなしていきます.13と14をやりました.ようやく実際にネットワークを組んで学習とテストを行います.簡単な深層学習まで辿り着きました!
Tutorial 13
「Feed Forward Neural Network - PyTorch Beginner 13」では,MNISTデータセットを使って手書き文字認識を行う.ということで,これまでやったことを一通りおさらいする感じでした.GPUでの実行も行いましたが,単にto("cuda")で必要なデータをGPUにアップロードするだけなので簡単ですね.
Tutorial 14
「Convolutional Neural Network (CNN) - PyTorch Beginner 14」では,ついにCNNを実装します.ただ,主に変わったところはネットワークの構造の中にnn.Conv2dとnn.MaxPool2dが入ってきて,チャンネル数を前後の層で揃えるくらい.
この程度のネットワークの学習に,すでに数分はかかっている(体感)ようです.結構,遅いですね.
所感
チュートリアルも残すところ3つになりました.全部こなしたところで自由自在にDNNを操れるようにはなれないでしょうが,既存のコードが読みやすくなることを期待しています.ちなみに,前回にVS Codeをセットアップしたおかげでスイスイ進みます.