Description
Filter and neuron pruning do not have add_mask_to_module() support so there is a NotImplementedError raised here or here. I was thinking I could catch and handle this exception since the masks aren't a necessity, but raising the exception breaks out of the for loop iterating over the layers in the scheme composer and leaves the remaining layers dense.
I am pretty sure saving the masks in the layer module isn't a necessity, so raising an exception seems like overkill because it stops the model from being pruned. You could also handle the exception inside the scheme composer's for loop so that it doesn't break out and is able to prune all the layers.
I can submit a PR myself if you want to describe which route you want to take for fixing this. Obviously supporting mask saving for filter and neuron pruning would also work, but I'm not sure what work needs to be done for that.