InternLM/internlm/core
jiaopenglong 9fc252f40e
add output embedding tf32 option (#523)
2023-12-06 13:50:59 +08:00
..
communication Merge develop to main (#233) 2023-08-24 22:03:04 +08:00
context feat(train): support_rampup_batch_size and fix bugs (#493) 2023-11-16 19:51:01 +08:00
scheduler 1. fix(config): rampup_batch_size defalut value BC. (#515) 2023-11-28 19:33:46 +08:00
__init__.py initial commit 2023-07-06 12:55:23 +08:00
engine.py initial commit 2023-07-06 12:55:23 +08:00
gradient_handler.py initial commit 2023-07-06 12:55:23 +08:00
naive_amp.py add output embedding tf32 option (#523) 2023-12-06 13:50:59 +08:00
trainer.py volc_path (#454) 2023-10-27 18:53:06 +08:00