07_03_Deploying_Mobile.ipynb

{"nbformat":4,"nbformat_minor":0,"metadata":{"colab":{"name":"07_03_Deploying_Mobile.ipynb","provenance":[],"collapsed_sections":[],"authorship_tag":"ABX9TyO+BuaqZhCEA4c8AoNHm8TP"},"kernelspec":{"name":"python3","display_name":"Python 3"}},"cells":[{"cell_type":"markdown","metadata":{"id":"Y2fkP_np3Zpl"},"source":["<!--BOOK_INFORMATION-->\n","<img align=\"left\" style=\"width:80px;height:98px;padding-right:20px;\" src=\"https://raw.githubusercontent.com/joe-papa/pytorch-book/main/files/pytorch-book-cover.jpg\">\n","\n","This notebook contains an excerpt from the [PyTorch Pocket Reference](http://pytorchbook.com) book by [Joe Papa](http://joepapa.ai); content is available [on GitHub](https://github.com/joe-papa/pytorch-book)."]},{"cell_type":"markdown","metadata":{"id":"tgEfXPZlIaE7"},"source":["[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/joe-papa/pytorch-book/blob/main/07_03_Deploying_Mobile.ipynb)"]},{"cell_type":"markdown","metadata":{"id":"aXkszegKMMMn"},"source":["# Chapter 7 - Deploy to Production"]},{"cell_type":"markdown","metadata":{"id":"g4Swe6JQS0J-"},"source":["### Mobile"]},{"cell_type":"code","metadata":{"id":"BxbRwzx3Osin"},"source":["import torch\n","import torchvision\n","from torch.utils.mobile_optimizer \\\n","  import optimize_for_mobile\n","\n","model = torchvision.models.vgg16(pretrained=True)\n","model.eval()\n","example = torch.rand(1, 3, 224, 224) # <1>\n","\n","traced_script_module = \\\n","  torch.jit.trace(model, example) # <2>\n","torchscript_model_optimized = \\\n","  optimize_for_mobile(traced_script_module) # <3>\n","torchscript_model_optimized.save(\"model.pt\") # <4>"],"execution_count":null,"outputs":[]},{"cell_type":"code","metadata":{"id":"GQpFKR1Si56D"},"source":[""],"execution_count":null,"outputs":[]}]}