LiBai v0.1.0 Release
Some major features:
- Data Parallelism
- 1D Tensor Parallelism
- Pipeline Parallelism
- Unified Distributed Layers for single-GPU and multi-GPU
- "LazyConfig" system for more flexible syntax and no predefined structures
- Easy-to-use trainer and engine
- CV & NLP Data Preprocessing
- Mixed Precision Training
- Activation Checkpointing
- Gradient Accumulation
- Gradient Clipping
- Zero Redundancy Optimizer (ZeRO)
Model_Zoo
- Bert (3D Parallelism)
- GPT-2 (3D Parallelism)
- ViT (3D Parallelism)
- Swin-Transformer (data parallelism)
- Add finetune task in
projects/
- Add text classification in
projects