search

CSAILVision / semantic-segmentation-pytorch

Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset
http://sceneparsing.csail.mit.edu/
BSD 3-Clause "New" or "Revised" License
4.96k stars 1.1k forks source link

summary of layers #269

Open ranch-hands opened 3 years ago

ranch-hands commented 3 years ago

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