Pytorchでモデル構築するとき、torchsummaryがマジ使える件について - Qiita

#はじめに自分でモデルを構築していて、いつも全結合層につなぐ前に「あれ、インプットの特徴量っていくつだ?」ってなります。よくprint(model)と打つとモデルの構造は理解できるが、Featur…