dhruvbird/OptimizeModelForMobileInference.py

Created November 12, 2021 19:27

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/d8a04bddb70422202a2eb25f17c916df.js"></script>
Save dhruvbird/d8a04bddb70422202a2eb25f17c916df to your computer and use it in GitHub Desktop.

Download ZIP

Optimize scripted model for mobile inference

Raw

OptimizeModelForMobileInference.py

	from torch.utils.mobile_optimizer import optimize_for_mobile
	optimized_model = optimize_for_mobile(scripted)

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