Skip to content

Instantly share code, notes, and snippets.

@aribornstein

aribornstein/Export Model.py

Created February 24, 2021 09:36
Show Gist options
  • Save aribornstein/da2f66e313507eb070f59667c05e5014 to your computer and use it in GitHub Desktop.
Save aribornstein/da2f66e313507eb070f59667c05e5014 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Export Model.py
# Export To Onnx
model.to_onnx(filepath, input_sample, export_params=True)
# Export to Torch Script
torch.jit.save(model.to_torchscript(), "model.pt")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment