PyTorchで作ったCNNでMNISTを識別する途中過程の画像を表示する簡単サンプル

PyTorchで作ったCNNでMNISTを識別する過程の途中経過

ちょっと気になったんで作ってみました。

ソースコード

ネットワーク(Netクラス)の各層で生成される戻り値を、クラスメンバに保存。
そのクラスメンバの内容をmatplotでビジュアライズすることで表示しています。

識別で使っている"mnist_cnn.pt"は、このコードで作れます。