jiangmingyan e871e342b3 [API] add docstrings and initialization to apex amp, naive amp (#3783) ...

* [mixed_precison] add naive amp demo

* [mixed_precison] add naive amp demo

* [api] add docstrings and initialization to apex amp, naive amp

* [api] add docstring to apex amp/ naive amp

* [api] add docstring to apex amp/ naive amp

* [api] add docstring to apex amp/ naive amp

* [api] add docstring to apex amp/ naive amp

* [api] add docstring to apex amp/ naive amp

* [api] add docstring to apex amp/ naive amp

* [api] fix

* [api] fix

2023-05-23 15:17:24 +08:00

..

mixed_precision

[API] add docstrings and initialization to apex amp, naive amp (#3783)

2023-05-23 15:17:24 +08:00

plugin

[plugin] a workaround for zero plugins' optimizer checkpoint (#3780)

2023-05-19 19:42:31 +08:00

__init__.py

[booster] implemented the torch ddd + resnet example (#3232)

2023-03-27 10:24:14 +08:00

accelerator.py

[booster] added the accelerator implementation (#3159)

2023-03-20 13:59:24 +08:00

booster.py

fixed the example docstring for booster (#3795)

2023-05-22 18:10:06 +08:00