v0.2.0
版本发布时间: 2023-08-24 22:07:11
InternLM/InternLM最新发布版本:v0.2.1dev20240102(2024-01-02 13:05:11)
Features:
- Support pipeline parallel, including interleaved and non-interleaved pipeline scheduler.
- Support sequence parallel.
- Support model evaluation.
- Support tf32 with flash-attention.
- Support tensorboard writer for recording training performance metrics.
- Support customed uniscale logger.
- Support calculating model's accuracy and perplexity metrics.
- Support oss storage and checkpoint asynchronous uploading.
- Support automatically loading the latest checkpoint.
- Support checkpoint snapshot.
- Support monitoring the status of training jobs, and alarm abnormal status.
- Support torch profiler.
- Support simple memory profiler.
Optimizations:
- Overlapping optimizer parameters broadcast with model forward.
- Overlapping optimizer last bucket gradients allreduce with compute norm.