more stats

Author: xadupre

_scripts/compare_model_execution.py

@@ -65,15 +65,15 @@
 print(f"-- creating session 1 from {onnx1!r}")
 opts = onnxruntime.SessionOptions()
 opts.optimized_model_filepath = "debug1_full.onnx"
+opts.log_severity_level = 0
+opts.log_verbosity_level = 0
 sess1 = onnxruntime.InferenceSession(onnx1, opts, providers=providers)
 print(f"-- creating session 2 from {onnx2!r}")
 opts.optimized_model_filepath = "debug2_full.onnx"
+opts.log_severity_level = 0
+opts.log_verbosity_level = 0
 sess2 = onnxruntime.InferenceSession(onnx2, opts, providers=providers)
 
-for n in ["debug1_full.onnx", "debug2_full.onnx"]:
-    x = onnx.load(n, load_external_data=False)
-    onnx.save(x, n.replace(".onnx", "-ext.onnx"), save_as_external_data=True)
-
 print("-- run session1")
 expected1 = sess1.run(None, feeds)
 print(f"-- got {string_type(expected1, with_shape=True)}")

onnx_diagnostic/torch_models/validate.py

@@ -852,15 +852,24 @@ def node_iter(proto):
             raise NotImplementedError(f"Unexpected type={type(proto)}")
 
     counts: Dict[str, Union[float, int]] = {}
+    n_nodes = 0
+    n_nodes_nocst = 0
     for proto in node_iter(onx):
         if isinstance(proto, onnx.NodeProto):
             key = f"n_node_{proto.op_type}"
+            n_nodes += 1
+            if proto.op_type != "Constant":
+                n_nodes_nocst += 1
         else:
             key = f"n_node_initializer_{proto.data_type}"
 
         if key not in counts:
             counts[key] = 0
         counts[key] += 1
+
+    counts["n_node_nodes"] = n_nodes
+    counts["n_node_nodes_nocst"] = n_nodes_nocst
+    counts["n_node_functions"] = len(onx.functions)
     return counts