Patch ORTTrainer's compatibility with DeepSpeed #148
JingyaHuang
merged
26 commits into
main
from
ort-trainer-ds
Add missing dependencies
52dd3555
Fix missing dependencies
8e00f8d8
Replace deprecated
1f4fb982
Unwrap model to get ORTModule & add ds configs for further tests
37d8e93f
Merge main
be133412
Fix style
b110c1df
Fix import
ef1ade54
Validate compatibility with fairscale
683cfb5d
Merge with main
1b8b6abc
Merge branch 'main' into ort-trainer-ds
024bffda
Fix compatibility of ort inference with DeepSpeed
941c3a2b
Add warnings
7b95a8cd
Update optimum/onnxruntime/trainer.py
f263c139
Update optimum/onnxruntime/trainer.py
902635ca
Update optimum/onnxruntime/trainer.py
09c62c75
Update optimum/onnxruntime/trainer.py
55bdfa73
Update optimum/onnxruntime/trainer.py
fe0a3161
Update optimum/onnxruntime/trainer.py
f1b2d1f1
echarlaix
commented
on 2022-05-16
Fix undefined
57a9753c
Merge branch 'ort-trainer-ds' of https://github.com/huggingface/optim…
c6be6a05
Merge branch 'main' into ort-trainer-ds
6d234467
Modify the _export with the flag in transformers.onnx.export(not merg…
ebc333b8
Modify send to device
1947390b
echarlaix
approved these changes
on 2022-05-23
Merge branch 'main' into ort-trainer-ds
280a82cd
Add transformers version demand
9d4ab38f
Add assets for testing BF16
4dc035a8
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub