polish amp docstring (#616)

3 years ago · c5b488edf8
parent f69507dd22
commit c5b488edf8
1 changed files with 1 additions and 2 deletions
--- a/colossalai/amp/naive_amp/init.py
+++ b/colossalai/amp/naive_amp/init.py
@ -18,8 +18,7 @@ def convert_to_naive_amp(model: nn.Module, optimizer: Optimizer, amp_config):
        amp_config (:class:`colossalai.context.Config` or dict): configuration for naive mode amp.


-    The ``amp_config`` should contain parameters below:
-    :
+    The ``amp_config`` should contain parameters below::

        verbose (bool, optional): if set to `True`, will print debug info (Default: False).
        clip_grad_norm (float, optional): clip gradients with this global L2 norm (Default 0).