InternLM/internlm/train
jiaopenglong debb7e77b9
refactor grad norm profiling (#466)
2023-11-03 10:55:26 +08:00
..
__init__.py feat(train): add fsdp training option (#293) 2023-10-09 18:59:31 +08:00
training_internlm.py refactor grad norm profiling (#466) 2023-11-03 10:55:26 +08:00
utils.py fix(moe): remove norm&gate force sync (#448) 2023-11-01 11:29:55 +08:00