First commit of the DRN ECAL Mustache regression weights for UL18

README.md

@@ -1 +1,4 @@
-# RecoEcal-EgammaClusterProducers
+# RecoEcal-EgammaClusterProducers
+This repository contains the weights to apply the DRN based energy regression correction to the Mustache superclusters reconstructed in the CMS ECAL 
+ - models/MustacheEB and models/MustacheEE contain the weights and config for the EB and EE regressions. 
+ - Each of those contains config.pbtxt with the triton configuration information and 1/model.pt with the model information and weights (ie this is the DRN compiled into torchscript) 

models/MustacheEB/config.pbtxt

@@ -0,0 +1,27 @@
+name: "MustacheEB"
+platform: "pytorch_libtorch"
+max_batch_size: 0
+input [
+{
+    name: "x__0"
+    data_type: TYPE_FP32
+    dims: [-1, 4]
+},
+{
+    name: "batch__1"
+    data_type: TYPE_INT64
+    dims: [-1]
+},
+{ 
+    name: "graphx__2"
+    data_type: TYPE_FP32
+    dims: [-1]
+}
+]
+output [
+{
+    name: "dscb__0"
+    data_type: TYPE_FP32
+    dims: [-1, 6]
+}]
+

models/MustacheEE/config.pbtxt

@@ -0,0 +1,27 @@
+name: "MustacheEE"
+platform: "pytorch_libtorch"
+max_batch_size: 0
+input [
+{
+    name: "x__0"
+    data_type: TYPE_FP32
+    dims: [-1, 4]
+},
+{
+    name: "batch__1"
+    data_type: TYPE_INT64
+    dims: [-1]
+},
+{ 
+    name: "graphx__2"
+    data_type: TYPE_FP32
+    dims: [-1]
+}
+]
+output [
+{
+    name: "dscb__0"
+    data_type: TYPE_FP32
+    dims: [-1, 6]
+}]
+