onnxruntime/orttraining/orttraining/python/training/optim
2021-10-05 20:50:34 -07:00
..
__init__.py
config.py
fused_adam.py Implement FusedAdam for ORT adapted from DeepSpeed (#9266) 2021-10-05 20:50:34 -07:00
lr_scheduler.py
multi_tensor_apply.py Implement FusedAdam for ORT adapted from DeepSpeed (#9266) 2021-10-05 20:50:34 -07:00