InternLM/internlm/solver
huangting4201 b9202b12bc
feat(utils/writer.py): support writer add_scalars for writing dict data (#257)
* feat(utils/writer.py): support writer add_scalars interface for writing dict data

* feat(hybrid_zero_optim.py): change grad_norm_groups list to dict
2023-09-01 13:24:46 +08:00
..
optimizer feat(utils/writer.py): support writer add_scalars for writing dict data (#257) 2023-09-01 13:24:46 +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