github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

refactor code

pull/375/head
Qu Wenwen 2023-09-21 15:00:28 +08:00
parent 9665321745
commit 17bc5f562b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internlm/train/utils.py
View File

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