sebamenabar / Pytorch-IntermediateLayerGetter

Simple easy to use module to get the intermediate values/activations from chosen Pytorch submodules. Supports submodule annidation and sequential modules.
54Updated 4 years ago

Related projects

Alternatives and complementary repositories for Pytorch-IntermediateLayerGetter