nntoolbox.hooks.stats module

class nntoolbox.hooks.stats.OutputStatsHook(module: torch.nn.modules.module.Module)

Bases: nntoolbox.hooks.hooks.Hook

class nntoolbox.hooks.stats.OutputStatsHooks(ms: List[torch.nn.modules.module.Module])

Bases: nntoolbox.hooks.hooks.Hooks