refactor code

2023-09-21 15:00:28 +08:00 · 2023-09-21 15:00:28 +08:00 · 17bc5f562b
parent 9665321745
commit 17bc5f562b
1 changed files with 1 additions and 1 deletions
--- a/internlm/train/utils.py
+++ b/internlm/train/utils.py
@ -10,7 +10,7 @@ def split_params_into_different_groups_for_optimizer(param_groups: Tuple[Dict])

    Args:
        param_groups (Tuple[Dict]): The list of parameter groups to split
-        Output Example:
+        Input Example:
        >>> (
        >>>     {'name': 'default', 'params': [tensor], 'weight_decay' :xxx},
        >>>     ...,