Skip to content

Instantly share code, notes, and snippets.

@jamesr66a
Created August 14, 2018 21:37
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save jamesr66a/229acab660154c715a57a7fb960295c8 to your computer and use it in GitHub Desktop.
Save jamesr66a/229acab660154c715a57a7fb960295c8 to your computer and use it in GitHub Desktop.
diff --git a/torch/onnx/utils.py b/torch/onnx/utils.py
index b770b900c..e940537f1 100644
--- a/torch/onnx/utils.py
+++ b/torch/onnx/utils.py
@@ -245,6 +245,7 @@ def _export(model, args, f, export_params=True, verbose=False, training=False,
training, input_names,
output_names, operator_export_type,
example_outputs, propagate)
+ print(graph)
# TODO: Don't allocate a in-memory string for the protobuf
from torch.onnx.symbolic import _onnx_opset_version
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment