onnxruntime/onnxruntime/python
Chi Lo 8c3b59a026
Quantization calibration refactor (#6893)
* Code refactor

* Modify code to tackle OOM when calibrating on larget dataset

* Fix mismatch issue when setting keepdims on ReduceMin/ReduceMax

* Add COCO val 2017 annotation

* Fix mismatch issue when setting keepdims on ReduceMin/ReduceMax

* Fix bug of "No module named:onnxruntime.quantization.CalTableFlatBuffers"

* Check and install flatbuffers module

* Add script to donwload coco dataset image and refactor example

* Fix bug of "No module
named:onnxruntime.quantization.CalTableFlatBuffers"

* Add CalTableFaltBuffers as module

* Remove annotation, user can download by themselves.

* Uncommet code

* Add back instances_val2017.json

* Make sure flatbuffers installed when ORT is installed

* Refactor code to call coco api

* Enable FP16 for example
2021-03-19 01:09:11 -07:00
..
backend fix py packaging pipeline (#5038) 2020-09-03 09:32:10 -07:00
datasets replace invalid sample (#4567) 2020-07-21 23:51:17 -07:00
tools Quantization calibration refactor (#6893) 2021-03-19 01:09:11 -07:00
training Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
__init__.py Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
_ld_preload.py Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
_pybind_state.py Add robust dependency check for Python package (#6436) 2021-02-21 15:11:28 -08:00
dlpack_convertor.cc Post merge update for ORTModule 2021-03-16 20:11:59 -07:00
dlpack_convertor.h Post merge update for ORTModule 2021-03-16 20:11:59 -07:00
exported_symbols.lst
onnxruntime_inference_collection.py Post merge update for ORTModule 2021-03-16 20:11:59 -07:00
onnxruntime_pybind_exceptions.h refactor frontend (#3235) 2020-03-19 20:59:41 -07:00
onnxruntime_pybind_mlvalue.cc Support sharing of initializers between session via the Python API (#5407) 2020-10-09 20:26:28 -07:00
onnxruntime_pybind_mlvalue.h Support sharing of initializers between session via the Python API (#5407) 2020-10-09 20:26:28 -07:00
onnxruntime_pybind_state.cc Merge pull request #6973 from microsoft/thiagofc/merge-ortmodule-into-master 2021-03-17 10:30:06 -07:00
onnxruntime_pybind_state_common.cc Enable passing initial optimizer state while creating training session (#5869) 2020-12-08 21:20:51 -05:00
onnxruntime_pybind_state_common.h Update ORT model conversion script to support custom ops (#6701) 2021-02-17 12:52:39 +10:00
onnxruntime_validation.py Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
pybind.def
version_script.lds