Skip to content

Instantly share code, notes, and snippets.

@oborchers

oborchers/modified_model_args.py

Created April 3, 2021 15:10
Show Gist options
  • Save oborchers/cf94514a21f8639f77bcaf0f6261de5c to your computer and use it in GitHub Desktop.
Save oborchers/cf94514a21f8639f77bcaf0f6261de5c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
modified_model_args.py
del model_args["dynamic_axes"]["output_0"] # Delete unused output
del model_args["dynamic_axes"]["output_1"] # Delete unused output
model_args["dynamic_axes"]["sentence_embedding"] = {0: "batch"}
model_args["output_names"] = ["sentence_embedding"]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment