InternLM/internlm/core/context
Wenwen Qu 3c992a2101
fix(pipeline): fix interleave type assert and metrics error (#423)
* fix interleave type assert bug

* refactor code for assert

* fix is_no_pp_or_last_stage logic
2023-10-19 17:29:30 +08:00
..
__init__.py feat(train): add fsdp training option (#293) 2023-10-09 18:59:31 +08:00
parallel_context.py fix(pipeline): fix interleave type assert and metrics error (#423) 2023-10-19 17:29:30 +08:00
process_group_initializer.py feat(train): add fsdp training option (#293) 2023-10-09 18:59:31 +08:00
random.py initial commit 2023-07-06 12:55:23 +08:00