0x00-pl/pytorch_module_hook_example

example of calculate sparsity of pytorch Module.

using torch.nn.Module.register_forward_hook .