onnxruntime/orttraining
ashbhandare 60c772e2bc
Megatron checkpointing (#6293)
* Add bart fairseq run script

* Add frontend change to enable megatron

* Initial changes for checkpointing

* Megatron optim state loading, checkpoint aggregation, frontend distributed tests for H, D+H

* Add load_checkpoint changes

* Fix CI

* Cleanup

* Fix CI

* review comments

* review comments

* review comments:
2021-01-22 11:26:47 -08:00
..
orttraining Megatron checkpointing (#6293) 2021-01-22 11:26:47 -08:00
pytorch_frontend_examples Enable running the mnist_training sample without cuda (#6085) 2020-12-15 17:06:54 -08:00
tools Dockerfile to build onnxruntime with ROCm 4.0 2020-12-22 10:21:12 -08:00