Changes needed for SONIC version of new particleNet models and Higgs …

…Interaction Network. Same structure for SONIC as original versions of particlenet
cms-data · Oct 27, 2023 · fb5f896 · fb5f896
1 parent c5cda6c
commit fb5f896
Show file tree

Hide file tree

Showing 25 changed files with 809 additions and 0 deletions.
diff --git a/HiggsInteractionNet/V00/IN.onnx b/HiggsInteractionNet/V00/IN.onnx
diff --git a/HiggsInteractionNet/V00/IN.onnx b/HiggsInteractionNet/V00/IN.onnx
@@ -0,0 +1 @@
+../../models/higgsInteractionNet/1/model.onnx
diff --git a/ParticleNetFromMiniAODAK4/CHS/Central/particle-net.onnx b/ParticleNetFromMiniAODAK4/CHS/Central/particle-net.onnx
diff --git a/ParticleNetFromMiniAODAK4/CHS/Central/particle-net.onnx b/ParticleNetFromMiniAODAK4/CHS/Central/particle-net.onnx
@@ -0,0 +1 @@
+../../../models/particleNetFromMiniAODAK4CHSCentral/1/model.onnx
diff --git a/ParticleNetFromMiniAODAK4/CHS/Forward/particle-net.onnx b/ParticleNetFromMiniAODAK4/CHS/Forward/particle-net.onnx
diff --git a/ParticleNetFromMiniAODAK4/CHS/Forward/particle-net.onnx b/ParticleNetFromMiniAODAK4/CHS/Forward/particle-net.onnx
@@ -0,0 +1 @@
+../../../models/particleNetFromMiniAODAK4CHSForward/1/model.onnx
diff --git a/ParticleNetFromMiniAODAK4/PUPPI/Central/particle-net.onnx b/ParticleNetFromMiniAODAK4/PUPPI/Central/particle-net.onnx
diff --git a/ParticleNetFromMiniAODAK4/PUPPI/Central/particle-net.onnx b/ParticleNetFromMiniAODAK4/PUPPI/Central/particle-net.onnx
@@ -0,0 +1 @@
+../../../models/particleNetFromMiniAODAK4PuppiCentral/1/model.onnx
diff --git a/ParticleNetFromMiniAODAK4/PUPPI/Forward/particle-net.onnx b/ParticleNetFromMiniAODAK4/PUPPI/Forward/particle-net.onnx
diff --git a/ParticleNetFromMiniAODAK4/PUPPI/Forward/particle-net.onnx b/ParticleNetFromMiniAODAK4/PUPPI/Forward/particle-net.onnx
@@ -0,0 +1 @@
+../../../models/particleNetFromMiniAODAK4PuppiForward/1/model.onnx
diff --git a/ParticleNetFromMiniAODAK8/particle-net.onnx b/ParticleNetFromMiniAODAK8/particle-net.onnx
diff --git a/ParticleNetFromMiniAODAK8/particle-net.onnx b/ParticleNetFromMiniAODAK8/particle-net.onnx
@@ -0,0 +1 @@
+../models/particleNetFromMiniAODAK8/1/model.onnx
diff --git a/models/higgsInteractionNet/1/model.onnx b/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/model.onnx
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		../models/particleNetFromMiniAODAK8/1/model.onnx