mmengine/tests
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
..
data [Enhance] Substitute the environment variable in config file (#744) 2023-02-01 10:16:47 +08:00
test_config [Enhance] Substitute the environment variable in config file (#744) 2023-02-01 10:16:47 +08:00
test_data
test_dataset [Enhance] Add `ignore_keys` in ConcatDataset (#556) 2022-11-01 17:06:02 +08:00
test_device [Feature] Support engine with NPU backend. (#572) 2022-10-24 11:42:29 +08:00
test_dist
test_evaluator [Fix] Fix offline_evaluate index error (#630) 2022-11-18 01:17:12 +08:00
test_fileio [Fix] backend_args should not be modified by get_file_backend (#897) 2023-02-01 14:47:22 +08:00
test_hooks [Feature] Support ReduceOnPlateauParamScheduler(#819) 2023-01-16 11:39:03 +08:00
test_hub [Docs] Fix docstring and documentation used for hub.get_model (#659) 2022-11-02 10:16:56 +08:00
test_infer [Feature] Add BaseInferencer (#874) 2023-01-16 16:01:06 +08:00
test_logging [Fix] Support update np.ScalarType data in message_hub (#898) 2023-02-01 23:53:28 +08:00
test_model [Fix] Unloaded weights will not be initialized when using PretrainedIinit (#764) 2023-01-09 18:46:30 +08:00
test_optim [Feature] Add ApexOptimWrapper (#742) 2023-02-06 15:30:10 +08:00
test_registry [Feature] Registry supports import modules automatically. (#643) 2022-12-23 15:46:29 +08:00
test_runner [Enhance] Support non-scalar type metric value. (#827) 2023-01-12 20:28:55 +08:00
test_structures [Docs] Update docstring of structures (#840) 2022-12-21 20:07:18 +08:00
test_testing [Enhance] enhance runner test case (#631) 2022-11-21 11:54:05 +08:00
test_utils [Fix] Ignore the distributed tests for macOS platform (#821) 2022-12-27 16:18:46 +08:00
test_visualizer [Fix] Fix failed unit tests caused by the new released numpy (#838) 2022-12-20 11:00:47 +08:00