Mugichoko's blog

Mugichoko’s blog

しがない研究者のプログラミングを中心としたメモ書き.

今更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.Conv2dnn.MaxPool2dが入ってきて,チャンネル数を前後の層で揃えるくらい.

この程度のネットワークの学習に,すでに数分はかかっている(体感)ようです.結構,遅いですね.

所感

チュートリアルも残すところ3つになりました.全部こなしたところで自由自在にDNNを操れるようにはなれないでしょうが,既存のコードが読みやすくなることを期待しています.ちなみに,前回にVS Codeをセットアップしたおかげでスイスイ進みます.