You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/booster
wukong1992 3229f93e30
[booster] add warning for torch fsdp plugin doc (#3833)
2 years ago
..
mixed_precision
plugin [booster] add warning for torch fsdp plugin doc (#3833) 2 years ago
__init__.py
accelerator.py
booster.py [booster] torch fsdp fix ckpt (#3788) 2 years ago