semantic-segmentation-pytorch icon indicating copy to clipboard operation
semantic-segmentation-pytorch copied to clipboard

summary of layers

Open ranch-hands opened this issue 4 years ago • 0 comments

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

ranch-hands avatar Nov 22 '21 12:11 ranch-hands