torch-module-observer
- PyTorch
- hooks
- feature-maps
- open-source
Un petit utilitaire PyTorch open-source qui instrumente n'importe quel nn.Module avec des forward hooks pour extraire les activations et cartes de features intermédiaires — on cible une couche, on passe une entrée, on récupère sa réponse pour l'inspecter ou la visualiser. Ci-dessous : les cartes de features d'une couche superficielle d'un ResNet (le chien encore lisible par filtre) à côté des activations abstraites d'une couche profonde.