CSAILVision / NetDissect

Network Dissection http://netdissect.csail.mit.edu for quantifying interpretability of deep CNNs.
MIT License
447 stars 115 forks source link

How to process the first conv layer in network? #23

Open Gchinanty opened 3 years ago

Gchinanty commented 3 years ago

LAYERS is set to "features" in the default rundissect_pytorch.sh, and after run rundissect_pytorch.sh, I will see the visualized results about the all 256 units from probes/pytorch_alexnet_imagenet/features.html. My question is if I want to see the scores of the first conv layer in network(such as alexnet) with 64 units, what should I do. If I need to make some changes to LAYERS in rundissect_pytorch.sh or do something else? I would be appreciate if the author could clear my confusion. THANKS!