diff --git a/README.md b/README.md index b14ef6f..edb1207 100644 --- a/README.md +++ b/README.md @@ -1 +1,4 @@ -# RecoEcal-EgammaClusterProducers \ No newline at end of file +# 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) diff --git a/models/MustacheEB/1/model.pt b/models/MustacheEB/1/model.pt new file mode 100644 index 0000000..c7d2a09 Binary files /dev/null and b/models/MustacheEB/1/model.pt differ diff --git a/models/MustacheEB/config.pbtxt b/models/MustacheEB/config.pbtxt new file mode 100644 index 0000000..d5ef2f2 --- /dev/null +++ b/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] +}] + diff --git a/models/MustacheEE/1/model.pt b/models/MustacheEE/1/model.pt new file mode 100644 index 0000000..e720cde Binary files /dev/null and b/models/MustacheEE/1/model.pt differ diff --git a/models/MustacheEE/config.pbtxt b/models/MustacheEE/config.pbtxt new file mode 100644 index 0000000..61f083e --- /dev/null +++ b/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] +}] +