【深層学習】PyTorch torchvizによる可視化

PyTorchのネットワーク構造がどうなっているのか、可視化したい場合に便利なのが「torchviz」というライブラリです。

Graphvizのインストール

まず、「Graphviz」をHomebrewを使ってインストールする必要があります。

%brew install graphviz

torchvizのインストール

仮想環境に入り、「torchviz」をインストールします。

(仮想環境名)%pip install torchviz

あとは、Pythonコードを書くと、以下のようなネットワーク図を表示することができます。

このように、処理の流れが非常にわかりやすい図が得られます。