From c5b488edf8b5a054c0149a3ab5770993bb61960a Mon Sep 17 00:00:00 2001 From: ver217 Date: Fri, 1 Apr 2022 16:09:39 +0800 Subject: [PATCH] polish amp docstring (#616) --- colossalai/amp/naive_amp/__init__.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/colossalai/amp/naive_amp/__init__.py b/colossalai/amp/naive_amp/__init__.py index e2cfc5a4e..889786361 100644 --- 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).