From 17bc5f562b06ccdcd0f4bd9c75b05fb7f96dd983 Mon Sep 17 00:00:00 2001
From: Qu Wenwen <vinny_qu@163.com>
Date: Thu, 21 Sep 2023 15:00:28 +0800
Subject: [PATCH] refactor code

---
 internlm/train/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/internlm/train/utils.py b/internlm/train/utils.py
index 29f238a..b0cf952 100644
--- 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},
         >>>     ...,