Alan Blanchet
All projects

torch-module-observer

  • PyTorch
  • hooks
  • feature-maps
  • open-source

A small open-source PyTorch utility that taps any nn.Module with forward hooks to pull out intermediate activations and feature maps — point it at a layer, run an input, get that layer's response back to inspect or visualise. Below: a ResNet's shallow-layer feature maps (the dog still legible per filter) beside a deep layer's abstract activations.