dhruvbird/SaveModelForMobileInference.py

Created November 12, 2021 19:28

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/dhruvbird/f73574e89f5112f4c41db239613fc8d7.js"></script>
Save dhruvbird/f73574e89f5112f4c41db239613fc8d7 to your computer and use it in GitHub Desktop.

Download ZIP

Save the model for mobile inference using lite-interpreter format

Raw

SaveModelForMobileInference.py

optimized_model._save_for_lite_interpreter("AddTensorsModelOptimized.ptl")

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment