InternLM/internlm/solver
jiaopenglong 112c34ae09
feat(grad_norm): vocab grad norm profiling (#519)
* compute vocab grad norm && save pt

* add grad_norm profiling interval && refactor save grad norm

* fix ci test_pipeline
2023-12-06 13:52:42 +08:00
..
optimizer feat(grad_norm): vocab grad norm profiling (#519) 2023-12-06 13:52:42 +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 feat(solver): fix code to adapt to torch2.0 and provide docker images (#128) 2023-07-25 19:34:52 +08:00
pipeline_utils.py initial commit 2023-07-06 12:55:23 +08:00