pneumonia_net_summary

pneumonia_summary.txt

==========================================================================================
Layer (type:depth-idx)                   Output Shape              Param #
==========================================================================================
├─Sequential: 1-1                        [-1, 576, 8, 8]           --
|    └─ConvBNActivation: 2-1             [-1, 16, 128, 128]        (464)
|    └─InvertedResidual: 2-2             [-1, 16, 64, 64]          (744)
|    └─InvertedResidual: 2-3             [-1, 24, 32, 32]          (3,864)
|    └─InvertedResidual: 2-4             [-1, 24, 32, 32]          (5,416)
|    └─InvertedResidual: 2-5             [-1, 40, 16, 16]          (13,736)
|    └─InvertedResidual: 2-6             [-1, 40, 16, 16]          (57,264)
|    └─InvertedResidual: 2-7             [-1, 40, 16, 16]          (57,264)
|    └─InvertedResidual: 2-8             [-1, 48, 16, 16]          (21,968)
|    └─InvertedResidual: 2-9             [-1, 48, 16, 16]          (29,800)
|    └─InvertedResidual: 2-10            [-1, 96, 8, 8]            (91,848)
|    └─InvertedResidual: 2-11            [-1, 96, 8, 8]            (294,096)
|    └─InvertedResidual: 2-12            [-1, 96, 8, 8]            294,096
|    └─ConvBNActivation: 2-13            [-1, 576, 8, 8]           56,448
├─Sequential: 1-2                        [-1, 2]                   --
|    └─AdaptiveAvgPool2d: 2-14           [-1, 576, 1, 1]           --
|    └─Flatten: 2-15                     [-1, 576]                 --
|    └─Dropout: 2-16                     [-1, 576]                 --
|    └─Linear: 2-17                      [-1, 2]                   1,154
==========================================================================================
Total params: 928,162
Trainable params: 351,698
Non-trainable params: 576,464
Total mult-adds (M): 13.32
==========================================================================================
Input size (MB): 0.75
Forward/backward pass size (MB): 4.56
Params size (MB): 3.54
Estimated Total Size (MB): 8.85
==========================================================================================