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
..
test_optimizer	[Feature] Add ApexOptimWrapper (#742)	2023-02-06 15:30:10 +08:00
test_scheduler	[Feature] Support ReduceOnPlateauParamScheduler(#819)	2023-01-16 11:39:03 +08:00