Skip to content

Commit

Permalink
update benchmark.md (#318)
Browse files Browse the repository at this point in the history
* update benchmark.md

* refine benchmark.md

* update changelog and readme

* add throughput optimization in changelog

* add benchmark link in changelog
  • Loading branch information
CPFLAME committed Jul 8, 2022
1 parent 9d08c09 commit 63c74a6
Show file tree
Hide file tree
Showing 3 changed files with 149 additions and 163 deletions.
37 changes: 14 additions & 23 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -69,33 +69,24 @@ See LiBai's [documentation](https://libai.readthedocs.io/en/latest/index.html) f

## ChangeLog

**Beta 0.1.0** was released in 22/03/2022, the main features and supported models in **0.1.0** version are as follows:
**Beta 0.2.0** was released in 07/07/2022, the general changes in **0.2.0** version are as follows:

**Features:**
- Support Data Parallelism
- Support 1D Tensor Parallelism
- Support Pipeline Parallelism
- Unified distributed Layers for both single-GPU and multi-GPU training
- `LazyConfig` system for more flexible syntax and no predefined structures
- Easy-to-use trainer and engine
- Support both CV and NLP data processing
- Mixed Precision Training
- Activation Checkpointing
- Gradient Accumulation
- Gradient Clipping
- Zero Redundancy Optimizer (ZeRO)
- Support evaluation enabled and set `eval_iter`
- Support customized sampler in `config.py`
- Support rdma for pipeline-model-parallel
- Support multi fused kernel
- fused_scale_mask_softmax_dropout
- fused_scale_tril_softmax_mask_scale
- fused_self_attention in branch `libai_bench`
- User Experience Optimization
- Optimization for training throughput, see [benchmark](https://libai.readthedocs.io/en/latest/tutorials/get_started/Benchmark.html) for more details

**Supported Models:**
- Support 3D parallel [BERT](https://arxiv.org/abs/1810.04805) model
- Support 3D parallel [GPT-2](https://cdn.openai.com/better-language-models/language_models_are_unsupervised_multitask_learners.pdf) model
- Support 3D parallel [T5](https://arxiv.org/abs/1910.10683) model
- Support 3D parallel [Vision Transformer](https://arxiv.org/abs/2010.11929) model
- Support Data parallel [Swin Transformer](https://arxiv.org/abs/2103.14030) model
- Support finetune task in [QQP project](/projects/QQP/)
- Support text classification task in [text classification project](/projects/text_classification/)
- Support Pathways Language Model (PaLM) in [PaLM project](/projects/PaLM/)
- Support MoCo_v3 in [MOCOV3 project](/projects/MOCOV3/)
- (experimental) Support MAE in [MAE project](/projects/MAE/)
- Support 3D parallel [Roberta](https://arxiv.org/abs/1907.11692) model
- Support 2D parallel (data parallel + tensor model parallel) [SimCSE](https://arxiv.org/abs/2104.08821) model
- Support Data parallel (MAE)[https://arxiv.org/abs/2111.06377] model
- Support Data parallel (MOCOV3)[https://arxiv.org/abs/2104.02057] model

See [changelog](./changelog.md) for details and release history.

Expand Down
19 changes: 19 additions & 0 deletions docs/source/changelog.md
Original file line number Diff line number Diff line change
@@ -1,5 +1,24 @@
## Changelog

### v0.2.0 (07/07/2022)
**New Features:**
- Support evaluation enabled and set `eval_iter`
- Support customized sampler in `config.py`
- Support rdma for pipeline-model-parallel
- Support multi fused kernel
- fused_scale_mask_softmax_dropout
- fused_scale_tril_softmax_mask_scale
- fused_self_attention in branch `libai_bench`
- User Experience Optimization
- Optimization for training throughput, see [benchmark](https://libai.readthedocs.io/en/latest/tutorials/get_started/Benchmark.html) for more details

**New Supported Models:**

- Support 3D parallel [Roberta](https://arxiv.org/abs/1907.11692) model
- Support 2D parallel (data parallel + tensor model parallel) [SimCSE](https://arxiv.org/abs/2104.08821) model
- Support Data parallel (MAE)[https://arxiv.org/abs/2111.06377] model
- Support Data parallel (MOCOV3)[https://arxiv.org/abs/2104.02057] model

### v0.1.0 (22/03/2022)

**New Features:**
Expand Down
Loading

0 comments on commit 63c74a6

Please sign in to comment.