saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-06 00:03:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/onnxruntime/test/python
History
ashbhandare 7cebf76a46
Improve checkpointing for Zero stage 1 (#5478) 
* Initial running changes

* Checkpointing aggregation changes

* compare with older version

* initial cleanup

* Add zero test, minor fix

* Fix zero test, transform, formatting

* Review comments

* add more unit tests

* review comments

* Try fix CI

* Add additional check on just aggregation code

* Try fix ckpt gen

* Add pregenerated ckpt for CI, enable zero test in e2e

* Moving test to nightly, removing ckpt files

* Add tests to dist GPU CI

* Fix dist test

* Review comments

* Fix test
2020-12-07 09:16:01 -08:00
..
contrib_ops Add Trilu custom op (#4537) 2020-08-17 14:42:26 -07:00
helper.py Introduce training changes. 2020-03-11 14:39:03 -07:00
onnx_backend_test_series.py Remove nGraph Execution Provider (#5858) 2020-11-19 16:47:55 -08:00
onnxruntime_test_ort_trainer.py Improve checkpointing for Zero stage 1 (#5478) 2020-12-07 09:16:01 -08:00
onnxruntime_test_ort_trainer_with_mixed_precision.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_python.py Fix build.py bug which prevents running some unit tests (#5990) 2020-12-03 08:57:55 -08:00
onnxruntime_test_python_backend.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_python_backend_mlops.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_python_iobinding.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_python_keras.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_python_mlops.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_python_nuphar.py Some fixes to symbolic shape inference (#5642) 2020-10-30 19:28:47 -07:00
onnxruntime_test_python_symbolic_shape_infer.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_training_unit_tests.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_training_unittest_utils.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00