refactor: SAM

Author: kozistr

pytorch_optimizer/sam.py

@@ -57,9 +57,9 @@ def __init__(
         adaptive: bool = False,
         **kwargs,
     ):
-        """
+        """SAM
         :param params: PARAMETERS. iterable of parameters to optimize or dicts defining parameter groups
-        :param base_optimizer: Optimizer.
+        :param base_optimizer: Optimizer. base optimizer
         :param rho: float. size of the neighborhood for computing the max loss
         :param adaptive: bool. element-wise Adaptive SAM
         :param kwargs: Dict. parameters for optimizer.