summary of layers

[ranch-hands]

hi everyone
do you know how can I see the layers and parameters of this network?
I use torchsumary but I don't know where should I put summary to see the outputs, parameters, and layer details. look at my code:

from torchsummary import summary
summary(segmentation_module, (3, 224, 224))
print(segmentation_module)

it doesn't work by the way. pls help