optimum
Refactoring ORTTrainer tests
#206
Merged

Refactoring ORTTrainer tests #206

JingyaHuang merged 43 commits into main from refac-trainer-tests
JingyaHuang
JingyaHuang Add missing dependencies
52dd3555
JingyaHuang Fix missing dependencies
8e00f8d8
JingyaHuang Replace deprecated
1f4fb982
JingyaHuang Unwrap model to get ORTModule & add ds configs for further tests
37d8e93f
JingyaHuang Merge main
be133412
JingyaHuang Fix style
b110c1df
JingyaHuang Fix import
ef1ade54
JingyaHuang Validate compatibility with fairscale
683cfb5d
JingyaHuang Merge with main
1b8b6abc
JingyaHuang Merge branch 'main' into ort-trainer-ds
024bffda
JingyaHuang Fix compatibility of ort inference with DeepSpeed
941c3a2b
JingyaHuang Add warnings
7b95a8cd
JingyaHuang Update optimum/onnxruntime/trainer.py
f263c139
JingyaHuang Update optimum/onnxruntime/trainer.py
902635ca
JingyaHuang Update optimum/onnxruntime/trainer.py
09c62c75
JingyaHuang Update optimum/onnxruntime/trainer.py
55bdfa73
JingyaHuang Update optimum/onnxruntime/trainer.py
fe0a3161
JingyaHuang Update optimum/onnxruntime/trainer.py
f1b2d1f1
JingyaHuang Fix undefined
57a9753c
JingyaHuang Merge branch 'ort-trainer-ds' of https://github.com/huggingface/optim…
c6be6a05
JingyaHuang Merge branch 'main' into ort-trainer-ds
6d234467
JingyaHuang Modify the _export with the flag in transformers.onnx.export(not merg…
ebc333b8
JingyaHuang Modify send to device
1947390b
JingyaHuang Merge main
1e9ae53c
JingyaHuang Add decorator for ort training
b55eb75e
HuggingFaceDocBuilderDev
JingyaHuang Add basic integration tests
9f2ee6f2
JingyaHuang Hub test
615b3e85
JingyaHuang Merged with main
a1c83821
JingyaHuang Update hub tester with optimum utils
620085e4
JingyaHuang Add optuna test
7c0b9224
JingyaHuang Update run.sh
b27f2c93
JingyaHuang Resolve conflicts with main
42933327
JingyaHuang Merge with main
86cf824b
JingyaHuang Modify run.sh
87e639c8
JingyaHuang merge with main
be13f5a7
JingyaHuang Merge with main
f9adb886
JingyaHuang Add HP search tests
73aa8099
JingyaHuang Add optimizer choice tests
c617fd24
JingyaHuang Add Seq2Seq tests
f2b6a12c
JingyaHuang Modify CI for ORT training
b1b342ac
JingyaHuang JingyaHuang marked this pull request as ready for review 4 years ago
JingyaHuang JingyaHuang changed the title [WIP] Refactoring ORTTrainer tests Refactoring ORTTrainer tests 4 years ago
JingyaHuang JingyaHuang requested a review from echarlaix echarlaix 4 years ago
JingyaHuang JingyaHuang requested a review from regisss regisss 4 years ago
JingyaHuang JingyaHuang requested a review from michaelbenayoun michaelbenayoun 4 years ago
regisss
regisss approved these changes on 2022-07-01
JingyaHuang Fix fp16 args
e6b53cb1
echarlaix
echarlaix approved these changes on 2022-07-04
JingyaHuang Fix indent of ort fused optimizer mock
92cb3875
JingyaHuang Fix sigopt requirement condition
bd20f51d
JingyaHuang JingyaHuang merged 7a5fcf0f into main 4 years ago
JingyaHuang JingyaHuang deleted the refac-trainer-tests branch 4 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone