Merge pull request #53 from wpmccormack/onnx_patch_with_newPN_HiggsIN

Adding SONIC structure for new ParticleNet and Higgs IN models + ONNX model update
cms-data · Nov 3, 2023 · 9af440f · 9af440f
2 parents 56174a1 + 18a4e22
commit 9af440f
Show file tree

Hide file tree

Showing 33 changed files with 831 additions and 8 deletions.
diff --git a/HiggsInteractionNet/V00/modelfile b/HiggsInteractionNet/V00/modelfile
@@ -0,0 +1 @@
+../../models/higgsInteractionNet/1
diff --git a/ParticleNetFromMiniAODAK4/CHS/Central/modelfile b/ParticleNetFromMiniAODAK4/CHS/Central/modelfile
@@ -0,0 +1 @@
+../../../models/particleNetFromMiniAODAK4CHSCentral/1
diff --git a/ParticleNetFromMiniAODAK4/CHS/Forward/modelfile b/ParticleNetFromMiniAODAK4/CHS/Forward/modelfile
@@ -0,0 +1 @@
+../../../models/particleNetFromMiniAODAK4CHSForward/1
diff --git a/ParticleNetFromMiniAODAK4/PUPPI/Central/modelfile b/ParticleNetFromMiniAODAK4/PUPPI/Central/modelfile
@@ -0,0 +1 @@
+../../../models/particleNetFromMiniAODAK4PuppiCentral/1
diff --git a/ParticleNetFromMiniAODAK4/PUPPI/Forward/modelfile b/ParticleNetFromMiniAODAK4/PUPPI/Forward/modelfile
@@ -0,0 +1 @@
+../../../models/particleNetFromMiniAODAK4PuppiForward/1
diff --git a/ParticleNetFromMiniAODAK8/modelfile b/ParticleNetFromMiniAODAK8/modelfile
@@ -0,0 +1 @@
+../models/particleNetFromMiniAODAK8/1
diff --git a/HiggsInteractionNet/V00/IN.onnx → models/higgsInteractionNet/1/model.onnx b/HiggsInteractionNet/V00/IN.onnx → models/higgsInteractionNet/1/model.onnx
diff --git a/models/higgsInteractionNet/config.pbtxt b/models/higgsInteractionNet/config.pbtxt
@@ -0,0 +1,27 @@
+name: "higgsInteractionNet"
+platform: "onnxruntime_onnx"
+max_batch_size : 500
+dynamic_batching {
+   preferred_batch_size: [ 200 ]
+}
+input [
+  {
+    name: "input_cpf"
+    data_type: TYPE_FP32
+    dims: [ 30, 60 ]
+  },
+  {
+    name: "input_sv"
+    data_type: TYPE_FP32
+    dims: [ 14, 5 ]
+  }
+]
+output [
+  {
+    name: "output1"
+    data_type: TYPE_FP32
+    dims: [ 2 ]
+    label_filename: "higgs_interactionnet_labels.txt"
+  }
+]
+optimization {graph : {level : -1}}
diff --git a/models/higgsInteractionNet/higgs_interactionnet_labels.txt b/models/higgsInteractionNet/higgs_interactionnet_labels.txt
@@ -0,0 +1,2 @@
+QCD
+Hbb
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		../../../models/particleNetFromMiniAODAK4CHSCentral/1