2356 Commits (8c62e50dbb0720e5b51984d94d0a28e1bb71fe7f)
 

Author SHA1 Message Date
Mingyan Jiang 8c62e50dbb [doc] update amp document 2 years ago
digger yu 9265f2d4d7
[NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779) 2 years ago
jiangmingyan e871e342b3
[API] add docstrings and initialization to apex amp, naive amp (#3783) 2 years ago
Frank Lee 615e2e5fc1
[test] fixed lazy init test import error (#3799) 2 years ago
Frank Lee ad93c736ea
[workflow] enable testing for develop & feature branch (#3801) 2 years ago
jiangmingyan ef02d7ef6d
[doc] update gradient accumulation (#3771) 2 years ago
Frank Lee f5c425c898
fixed the example docstring for booster (#3795) 2 years ago
Frank Lee 788e07dbc5
[workflow] fixed the docker build workflow (#3794) 2 years ago
liuzeming 4d29c0f8e0
Fix/docker action (#3266) 2 years ago
github-actions[bot] 62c7e67f9f
[format] applied code formatting on changed files in pull request 3786 (#3787) 2 years ago
jiangmingyan fe1561a884
[doc] update gradient cliping document (#3778) 2 years ago
Yanjia0 d9393b85f1
[doc] add deprecated warning on doc Basics section (#3754) 2 years ago
Hongxin Liu 72688adb2f
[doc] add booster docstring and fix autodoc (#3789) 2 years ago
Hongxin Liu 3c07a2846e
[plugin] a workaround for zero plugins' optimizer checkpoint (#3780) 2 years ago
Hongxin Liu 60e6a154bc
[doc] add tutorial for booster checkpoint (#3785) 2 years ago
binmakeswell ad2cf58f50
[chat] add performance and tutorial (#3786) 2 years ago
Hongxin Liu b4788d63ed
[devops] fix doc test on pr (#3782) 2 years ago
digger yu 32f81f14d4
[NFC] fix typo colossalai/amp auto_parallel autochunk (#3756) 2 years ago
Hongxin Liu 21e29e2212
[doc] add tutorial for booster plugins (#3758) 2 years ago
Hongxin Liu 5ce6c9d86f
[doc] add tutorial for cluster utils (#3763) 2 years ago
Hongxin Liu 5452df63c5
[plugin] torch ddp plugin supports sharded model checkpoint (#3775) 2 years ago
jiangmingyan 2703a37ac9
[amp] Add naive amp demo (#3774) 2 years ago
jiangmingyan 48bd056761
[doc] update hybrid parallelism doc (#3770) 2 years ago
binmakeswell 15024e40d9
[auto] fix install cmd (#3772) 2 years ago
jiangmingyan d449525acf
[doc] update booster tutorials (#3718) 2 years ago
Yuanchen 05759839bd
[chat] fix bugs in stage 3 training (#3759) 2 years ago
Hongxin Liu 5dd573c6b6
[devops] fix ci for document check (#3751) 2 years ago
Hongxin Liu c03bd7c6b2
[devops] make build on PR run automatically (#3748) 2 years ago
digger yu 1baeb39c72
[NFC] fix typo with colossalai/auto_parallel/tensor_shard (#3742) 2 years ago
Ziyue Jiang 7386c6669d
[fix] Add init to fix import error when importing _analyzer (#3668) 2 years ago
wukong1992 6050f37776
[booster] removed models that don't support fsdp (#3744) 2 years ago
Hongxin Liu afb239bbf8
[devops] update torch version of CI (#3725) 2 years ago
wukong1992 b37797ed3d
[booster] support torch fsdp plugin in booster (#3697) 2 years ago
digger-yu ad6460cf2c
[NFC] fix typo applications/ and colossalai/ (#3735) 2 years ago
digger-yu 1f73609adb
[CI] fix typo with tests/ etc. (#3727) 2 years ago
digger-yu 899aa86368
[CI] fix typo with tests components (#3695) 2 years ago
digger-yu b7141c36dd
[CI] fix some spelling errors (#3707) 2 years ago
MisterLin1995 f7361ee1bd
[chat] fix community example ray (#3719) 2 years ago
jiangmingyan 20068ba188
[booster] add tests for ddp and low level zero's checkpointio (#3715) 2 years ago
Hongxin Liu 6552cbf8e1
[booster] fix no_sync method (#3709) 2 years ago
Hongxin Liu 3bf09efe74
[booster] update prepare dataloader method for plugin (#3706) 2 years ago
Hongxin Liu f83ea813f5
[example] add train resnet/vit with booster example (#3694) 2 years ago
YH 2629f9717d
[tensor] Refactor handle_trans_spec in DistSpecManager 2 years ago
zhang-yi-chi 2da5d81dec
[chat] fix train_prompts.py gemini strategy bug (#3666) 2 years ago
Hongxin Liu d556648885
[example] add finetune bert with booster example (#3693) 2 years ago
digger-yu 65bdc3159f
fix some spelling error with applications/Chat/examples/ (#3692) 2 years ago
Hongxin Liu d0915f54f4
[booster] refactor all dp fashion plugins (#3684) 2 years ago
digger-yu b49020c1b1
[CI] Update test_sharded_optim_with_sync_bn.py (#3688) 2 years ago
Tong Li b36e67cb2b
Merge pull request #3680 from digger-yu/digger-yu-patch-2 2 years ago
jiangmingyan 307894f74d
[booster] gemini plugin support shard checkpoint (#3610) 2 years ago