[skip ci] docs: SAM docstring

Author: kozistr

pytorch_optimizer/optimizer/sam.py

@@ -11,7 +11,7 @@
 class SAM(Optimizer, BaseOptimizer):
     r"""Sharpness-Aware Minimization for Efficiently Improving Generalization
 
-    Example :
+    Example:
         from pytorch_optimizer import SAM
         ...
         model = YourModel()
@@ -50,11 +50,13 @@ def closure():
             optimizer.step(closure)
             optimizer.zero_grad()
 
-    :param params: PARAMETERS. iterable of parameters to optimize or dicts defining parameter groups
-    :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.
+    Arguments:
+
+        params: PARAMETERS. iterable of parameters to optimize or dicts defining parameter groups
+        base_optimizer: Optimizer. base optimizer
+        rho: float. size of the neighborhood for computing the max loss
+        adaptive: bool. element-wise Adaptive SAM
+        kwargs: Dict. parameters for optimizer.
     """
 
     def __init__(