InternLM/internlm/solver
ytxiong d67be17f96
refactor(*): refactor the code with no-apex (#170)
* support no-apex

* add default for use_apex

* fix lint

* modify the RMSNormTorch

* remove some comments

* remove use_apex parameter

* remove some unnecessary code

* optimize the code including import

* remove the import RMSNorm

* remove warnings
2023-08-03 11:24:12 +08:00
..
optimizer refactor(*): refactor the code with no-apex (#170) 2023-08-03 11:24:12 +08:00
__init__.py initial commit 2023-07-06 12:55:23 +08:00
beta2_scheduler.py initial commit 2023-07-06 12:55:23 +08:00
lr_scheduler.py [Dev] Pull Main (#139) 2023-07-27 10:20:21 +08:00
pipeline_utils.py initial commit 2023-07-06 12:55:23 +08:00