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
_apex_amp_modifier.py
_ds_modifier.py [ORTModule] Update Supported DeepSpeed Version for FP16_Optimizer (#13305) 2022-10-13 13:03:01 +08:00
_megatron_modifier.py
_modifier.py
_modifier_registry.py
_multi_tensor_apply.py
config.py
fp16_optimizer.py Fix segment fault for alltoall (#12701) 2022-08-30 11:27:14 +08:00
fused_adam.py
lr_scheduler.py