Pytorchでモデル構築するとき、torchsummaryがマジ使える件に …
https://qiita.com/tatsuya11bbs/items/bfd8f163dd918ea0443221.01.2020 · 今回は、画像の入力サイズを224x224を想定して試しています。 他の解像度で試したいときは、HとWの値を変更してください。 summaryの出力-----Layer (type) Output Shape Param # ===== Conv2d-1 [-1, 16, 222, 222] 448 BatchNorm2d-2 [-1, 16, 222, 222] 32 ReLU-3 [-1, 16, 222, 222] 0 MaxPool2d-4 [-1, 16, 111, 111] 0 Conv2d-5 [-1, 32, 109, 109] 4, 640 ...