setter function for quant enable_on_initialize (#203)

* setter function for quant enable_on_initialize

* format docstring

* update docstring

Author: bfineran

src/sparseml/pytorch/optim/modifier_quantization.py

@@ -218,6 +218,14 @@ def enable_on_initialize(self) -> bool:
         """
         return self._enable_on_initialize
 
+    @enable_on_initialize.setter
+    def enable_on_initialize(self, value: bool):
+        """
+        :params value: True if QAT should be enabled when this modifier is initialized,
+            False otherwise. If true, start_epoch will not be respected
+        """
+        self._enable_on_initialize = value
+
     def initialize(self, module: Module, optimizer: Optimizer):
         """
         Grab the module / submodule to perform QAT on