How to grab other layers once prediction is done?

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

Closed joshmyersdean closed 5 years ago

joshmyersdean commented 5 years ago

I'm having a bit of trouble grabbing the last convolution layer from the model after the data has been ran through.