onnxruntime/onnxruntime
ashbhandare 1c316d0e39
Parameter,Module and Optimizer changes (#11494)
* Module step

* On device training offline composition

* Working grad accumulation with test for TrainStep

* Temp changes

* Revert "On device training offline composition"

This reverts commit ec3da68247.

* cleanup

* Implement eval step

* Use new graphs and checkpoints

* Optimizer test, changes

* review comments

* review comments

* review comments

Co-authored-by: Aishwarya Bhandare <aibhanda@microsoft.com@orttrainingdev7.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net>
Co-authored-by: Baiju Meswani <bmeswani@microsoft.com>
2022-05-31 09:20:47 -07:00
..
contrib_ops [CUDA] Implement BitmaskDropout, BitmaskBiasDropout and BitmaskDropoutGrad (#11534) 2022-05-27 17:24:47 +08:00
core Parameter,Module and Optimizer changes (#11494) 2022-05-31 09:20:47 -07:00
gsl
python [trt-ep-perf] Fix upload time of EP perf data (#11531) 2022-05-18 15:36:21 -07:00
test Parameter,Module and Optimizer changes (#11494) 2022-05-31 09:20:47 -07:00
tool/etw
wasm
__init__.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
ReformatSource.ps1
ReformatSourcePython.bat Add python docstring linting in vscode settings (#11316) 2022-04-23 06:23:04 -07:00
VSCodeCoverage.runsettings