TorchInfo should say the size of the model, right?

1

I am not sure why TorchInfo is not reporting the size of the model.
My code looks like this
import torchinfo from torchinfo import summary summary(model_0, input_size=print(img_custom.shape)) # do a test pass through of an example input size
The model output looks like this:

Layer (type:depth-idx) Param #

TinyVGG –
├─Sequential: 1-1 –
│ └─Conv2d: 2-1 11,530
│ └─ReLU: 2-2 –
│ └─Conv2d: 2-3 910
│ └─ReLU: 2-4 –
│ └─MaxPool2d: 2-5 –
├─Sequential: 1-2 –
│ └─Conv2d: 2-6 910
│ └─ReLU: 2-7 –
│ └─Conv2d: 2-8 910
│ └─ReLU: 2-9 –
│ └─MaxPool2d: 2-10 –
├─Sequential: 1-3 –
│ └─Flatten: 2-11 –
│ └─Linear: 2-12 94,083

Total params: 108,343
Trainable params: 108,343
Non-trainable params: 0