onnxruntime/orttraining/orttraining/python/training/optim
Vincent Wang 6fb70a82df
[ORTModule] Update Supported DeepSpeed Version for FP16_Optimizer (#13305)
Update supported deepspeed highest version from 0.7.1 to 0.7.3 for
FP16_Optimizer. Also add version info to warning log.
2022-10-13 13:03:01 +08:00
..
__init__.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
_apex_amp_modifier.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
_ds_modifier.py [ORTModule] Update Supported DeepSpeed Version for FP16_Optimizer (#13305) 2022-10-13 13:03:01 +08:00
_megatron_modifier.py Remove Apex Dependency For Deepspeed FP16_Optimizer (#12077) 2022-07-14 11:15:53 +08:00
_modifier.py Remove Apex Dependency For Deepspeed FP16_Optimizer (#12077) 2022-07-14 11:15:53 +08:00
_modifier_registry.py FP16_Optimizer Support for more Deepspeed Versions (#12046) 2022-06-30 18:36:17 +08:00
_multi_tensor_apply.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
config.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
fp16_optimizer.py Fix segment fault for alltoall (#12701) 2022-08-30 11:27:14 +08:00
fused_adam.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
lr_scheduler.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00