Pythonいぬ

pythonを使った画像処理に関する記事を書いていきます

2020-04-18から1日間の記事一覧

Pytorchでkerasのsummaryをやる

kerasを使っていたときは、model.summary()という関数があって、ネットワークの各レイヤにおける出力サイズがどうなっていくかを簡単に可視化できていた。 Pytorchはdefine by runなのでネットワーク内の各層のサイズはforward処理のときに決まる。なのでな…