Skip to content

GitLab

Unverified Commit c5b488ed authored by ver217's avatar ver217 Committed by GitHub
Browse files

polish amp docstring (#616)

parent f69507dd
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
colossalai/amp/naive_amp/__init__.py
View file @ c5b488ed
...@@ -18,8 +18,7 @@ def convert_to_naive_amp(model: nn.Module, optimizer: Optimizer, amp_config): ...@@ -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. 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). 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). clip_grad_norm (float, optional): clip gradients with this global L2 norm (Default 0).
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment