Prune Callback
Use the pruner in fastai Callback system
PruneCallback
PruneCallback (sparsity:int, context:str, criteria:Callable, schedule:Callable, model:torch.nn.modules.module.Module=None, round_to:int=None, layer_type:torch.nn.modules.module.Module=<class 'torch.nn.modules.conv.Conv2d'>)
Basic class handling tweaks of the training loop by changing a Learner
in various events