dhruvbird/ScriptPyTorchModel.py

Created November 12, 2021 19:26

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

Download ZIP

Create a scripted model

Raw

ScriptPyTorchModel.py

scripted = torch.jit.script(m)

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