ml-utils

Helpful ML functions.

Dependencies

• Python 3.6.9
• Pip3

Installation

pip install .

Usage

from mlutils import check_onnx_model

PYTORCH_MODEL_PATH = path/to/model.pth

check_onnx_model(PYTORCH_MODEL_PATH)

Acknowledgements

While developing this package, I took help from the following sources:

• https://github.com/onnx/onnx/blob/master/docs/PythonAPIOverview.md
• https://pypi.org/project/onnx2keras/
• https://www.tensorflow.org/lite/convert
• https://pytorch.org/tutorials/advanced/super_resolution_with_onnxruntime.html

Questions?

Feel free to contact me at arun.etc.kumar@gmail.com if you have any questions.