Alex Yang ef946404e6
[Feat] Support FSDP Training (#304)
* [Feat] Support FSDP Training

* fix version comparison

* change param format and move `FSDP_WRAP_POLICY` to wrapper file

* add docstring and type hint,reformat code

* fix type hint

* fix typo, reformat code
2022-06-21 15:32:56 +08:00
2022-06-21 15:13:53 +08:00
2022-06-21 15:32:56 +08:00
2022-03-30 23:26:12 +08:00
2022-05-19 17:56:31 +08:00

MMEngine

Description
OpenMMLab Foundational Library for Training Deep Learning Models
Readme 20 MiB
Languages
Python 99.9%
Dockerfile 0.1%