move remaining torch pruning modifiers to /sparsification/pruning (#571)

* refactor movement and layer pruning modifiers + delete optim/modifier_pruning

* refactor note

* quality

* review fixes

* remove buggy test - to be fixed post release

Author: bfineran

src/sparseml/pytorch/optim/__init__.py

@@ -33,7 +33,6 @@
 from .modifier_epoch import *
 from .modifier_lr import *
 from .modifier_params import *
-from .modifier_pruning import *
 from .modifier_regularizer import *
 from .optimizer import *
 from .sensitivity_as import *

src/sparseml/pytorch/optim/mask_creator_pruning.py

@@ -14,6 +14,10 @@
 
 """
 Classes for defining sparsity masks based on model parameters.
+
+NOTE: this file is in the process of being phased out in favor of the
+sparsification package. Once all references to mask utils in the optim
+package are migrated, this file will be deleted
 """
 
 import random

src/sparseml/pytorch/optim/mask_pruning.py

@@ -15,6 +15,10 @@
 """
 Code related to applying a mask onto a parameter to impose kernel sparsity,
 aka model pruning
+
+NOTE: this file is in the process of being phased out in favor of the
+sparsification package. Once all references to mask utils in the optim
+package are migrated, this file will be deleted
 """
 
 from functools import partial

src/sparseml/pytorch/optim/mask_pruning_scorer.py

@@ -14,6 +14,10 @@
 
 """
 Classes for tracking and scoring model parameters to generate pruning scores
+
+NOTE: this file is in the process of being phased out in favor of the
+sparsification package. Once all references to mask utils in the optim
+package are migrated, this file will be deleted
 """