xcnick e35ed5fd2e

[Feature] Add ApexOptimWrapper (#742) ...

* add ApexOptimWrapper

* typo fix

* add apex amp.initialize in optim_context

* assert apex_amp

* polish code

* add parameters of apex_amp.initialize

* add docs

* polish code

* polish code

* polish code

* fix calling of apex amp load_state_dict

* polish

* add comments

* Update apex_optimizer_wrapper.py

* Update apex_optimizer_wrapper.py

---------

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

2023-02-06 15:30:10 +08:00

..

config.rst

[Docs] Make API of docs clearer (#469)

2022-08-30 16:27:20 +08:00

dataset.rst

[Docs] Fix API files of en docs (#525)

2022-09-13 14:57:18 +08:00

device.rst

[Feature] Support engine with NPU backend. (#572)

2022-10-24 11:42:29 +08:00

dist.rst

[Docs] Fix API files of en docs (#525)

2022-09-13 14:57:18 +08:00

evaluator.rst

[Docs] Fix API files of en docs (#525)

2022-09-13 14:57:18 +08:00

fileio.rst

[Refactor] Refactor fileio without breaking back compatibility (#533)

2022-09-26 14:30:40 +08:00

hooks.rst

Make TTAModel compatible with FSDP (#611)

2022-12-27 21:39:43 +08:00

hub.rst

[Docs] Fix API files of en docs (#525)

2022-09-13 14:57:18 +08:00

infer.rst

[Feature] Add BaseInferencer (#874)

2023-01-16 16:01:06 +08:00

logging.rst

[Docs] Fix API files of en docs (#525)

2022-09-13 14:57:18 +08:00

model.rst

[Feature] Add test time augmentation base model. (#538)

2022-10-10 14:29:33 +08:00

optim.rst

[Feature] Add ApexOptimWrapper (#742)

2023-02-06 15:30:10 +08:00

registry.rst

[Feature] Registry supports import modules automatically. (#643)

2022-12-23 15:46:29 +08:00

runner.rst

[Docs] Fix API files of en docs (#525)

2022-09-13 14:57:18 +08:00

structures.rst

[Docs] Make API of docs clearer (#469)

2022-08-30 16:27:20 +08:00

utils.dl_utils.rst

[Docs] Fix API files of en docs (#525)

2022-09-13 14:57:18 +08:00

utils.rst

[Feature] Add a function to mark the deprecated function. (#609)

2022-10-24 14:49:48 +08:00

visualization.rst

[Docs] Make API of docs clearer (#469)

2022-08-30 16:27:20 +08:00