nntoolbox.hooks package

Abstraction for pytorch hooks (similar to callbacks but for layers)