saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-31 23:27:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/orttraining/orttraining/python
History
Adam Louly fb4707f76d
add cuda support to python bindings (#13700) 
### Description
Add cuda support to the on device training python bindings.



### Motivation and Context
Now users can set the execution provider (cpu or cuda) when using python
bindings for on device training apis.
2022-12-08 16:03:53 -08:00
..
deprecated Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
training add cuda support to python bindings (#13700) 2022-12-08 16:03:53 -08:00
checkpointing_utils.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
ort_trainer.py Deprecate ORTTrainer (#13022) 2022-09-23 18:10:09 -07:00
orttraining_pybind_common.h
orttraining_pybind_state.cc add cuda support to python bindings (#13700) 2022-12-08 16:03:53 -08:00
orttraining_python_module.cc Allow CUDA EP enable or disable TunableOp via session options and environment variable (#13601) 2022-11-15 14:43:54 +08:00
orttraining_python_module_eager.h
pt_patch.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00