add model checksums

cms-data · Jan 3, 2024 · 1c5dbaa · 1c5dbaa
1 parent 9af440f
commit 1c5dbaa
Show file tree

Hide file tree

Showing 14 changed files with 676 additions and 628 deletions.
diff --git a/models/higgsInteractionNet/config.pbtxt b/models/higgsInteractionNet/config.pbtxt
@@ -1,27 +1,35 @@
 name: "higgsInteractionNet"
 platform: "onnxruntime_onnx"
-max_batch_size : 500
+max_batch_size: 500
+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"
+}]
 dynamic_batching {
-   preferred_batch_size: [ 200 ]
+  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 ]
+optimization {
+  graph {
+    level: -1
   }
-]
-output [
-  {
-    name: "output1"
-    data_type: TYPE_FP32
-    dims: [ 2 ]
-    label_filename: "higgs_interactionnet_labels.txt"
-  }
-]
-optimization {graph : {level : -1}}
+}
+model_repository_agents {
+  agents [{
+    name: "checksum"
+    parameters {
+      key: "MD5:1/model.onnx"
+      value: "c245c8227157b4098c1c8e2decbdd3d5"
+    }
+  }]
+}
diff --git a/models/particleNetFromMiniAODAK4CHSCentral/config.pbtxt b/models/particleNetFromMiniAODAK4CHSCentral/config.pbtxt
@@ -1,62 +1,63 @@
 name: "particleNetFromMiniAODAK4CHSCentral"
 platform: "onnxruntime_onnx"
-max_batch_size : 500
+max_batch_size: 500
+input [{
+  name: "pf_points"
+  data_type: TYPE_FP32
+  dims: [2, -1]
+}, {
+  name: "pf_features"
+  data_type: TYPE_FP32
+  dims: [41, -1]
+}, {
+  name: "pf_mask"
+  data_type: TYPE_FP32
+  dims: [1, -1]
+}, {
+  name: "sv_points"
+  data_type: TYPE_FP32
+  dims: [2, -1]
+}, {
+  name: "sv_features"
+  data_type: TYPE_FP32
+  dims: [10, -1]
+}, {
+  name: "sv_mask"
+  data_type: TYPE_FP32
+  dims: [1, -1]
+}, {
+  name: "lt_points"
+  data_type: TYPE_FP32
+  dims: [2, -1]
+}, {
+  name: "lt_features"
+  data_type: TYPE_FP32
+  dims: [19, -1]
+}, {
+  name: "lt_mask"
+  data_type: TYPE_FP32
+  dims: [1, -1]
+}]
+output [{
+  name: "output"
+  data_type: TYPE_FP32
+  dims: [20]
+  label_filename: "particlenet_labels.txt"
+}]
 dynamic_batching {
-   preferred_batch_size: [ 200 ]
+  preferred_batch_size: [200]
 }
-input [
-  {
-    name: "pf_points"
-    data_type: TYPE_FP32
-    dims: [ 2, -1 ]
-  },
-  {
-    name: "pf_features"
-    data_type: TYPE_FP32
-    dims: [ 41, -1 ]
-  },
-  {
-    name: "pf_mask"
-    data_type: TYPE_FP32
-    dims: [ 1, -1 ]
-  },
-  {
-    name: "sv_points"
-    data_type: TYPE_FP32
-    dims: [ 2, -1 ]
-  },
-  {
-    name: "sv_features"
-    data_type: TYPE_FP32
-    dims: [ 10, -1 ]
-  },
-  {
-    name: "sv_mask"
-    data_type: TYPE_FP32
-    dims: [ 1, -1 ]
-  },
-  {
-    name: "lt_points"
-    data_type: TYPE_FP32
-    dims: [ 2, -1 ]
-  },
-  {
-    name: "lt_features"
-    data_type: TYPE_FP32
-    dims: [ 19, -1 ]
-  },
-  {
-    name: "lt_mask"
-    data_type: TYPE_FP32
-    dims: [ 1, -1 ]
+optimization {
+  graph {
+    level: -1
   }
-]
-output [
-  {
-    name: "output"
-    data_type: TYPE_FP32
-    dims: [ 20 ]
-    label_filename: "particlenet_labels.txt"
-  }
-]
-optimization {graph : {level : -1}}
+}
+model_repository_agents {
+  agents [{
+    name: "checksum"
+    parameters {
+      key: "MD5:1/model.onnx"
+      value: "3e08eabde2e40e318d2f5f794cc288d7"
+    }
+  }]
+}
diff --git a/models/particleNetFromMiniAODAK4CHSForward/config.pbtxt b/models/particleNetFromMiniAODAK4CHSForward/config.pbtxt
@@ -1,62 +1,63 @@
 name: "particleNetFromMiniAODAK4CHSForward"
 platform: "onnxruntime_onnx"
-max_batch_size : 500
+max_batch_size: 500
+input [{
+  name: "pf_points"
+  data_type: TYPE_FP32
+  dims: [2, -1]
+}, {
+  name: "pf_features"
+  data_type: TYPE_FP32
+  dims: [40, -1]
+}, {
+  name: "pf_mask"
+  data_type: TYPE_FP32
+  dims: [1, -1]
+}, {
+  name: "sv_points"
+  data_type: TYPE_FP32
+  dims: [2, -1]
+}, {
+  name: "sv_features"
+  data_type: TYPE_FP32
+  dims: [10, -1]
+}, {
+  name: "sv_mask"
+  data_type: TYPE_FP32
+  dims: [1, -1]
+}, {
+  name: "lt_points"
+  data_type: TYPE_FP32
+  dims: [2, -1]
+}, {
+  name: "lt_features"
+  data_type: TYPE_FP32
+  dims: [19, -1]
+}, {
+  name: "lt_mask"
+  data_type: TYPE_FP32
+  dims: [1, -1]
+}]
+output [{
+  name: "output"
+  data_type: TYPE_FP32
+  dims: [6]
+  label_filename: "particlenet_labels.txt"
+}]
 dynamic_batching {
-   preferred_batch_size: [ 200 ]
+  preferred_batch_size: [200]
 }
-input [
-  {
-    name: "pf_points"
-    data_type: TYPE_FP32
-    dims: [ 2, -1 ]
-  },
-  {
-    name: "pf_features"
-    data_type: TYPE_FP32
-    dims: [ 40, -1 ]
-  },
-  {
-    name: "pf_mask"
-    data_type: TYPE_FP32
-    dims: [ 1, -1 ]
-  },
-  {
-    name: "sv_points"
-    data_type: TYPE_FP32
-    dims: [ 2, -1 ]
-  },
-  {
-    name: "sv_features"
-    data_type: TYPE_FP32
-    dims: [ 10, -1 ]
-  },
-  {
-    name: "sv_mask"
-    data_type: TYPE_FP32
-    dims: [ 1, -1 ]
-  },
-  {
-    name: "lt_points"
-    data_type: TYPE_FP32
-    dims: [ 2, -1 ]
-  },
-  {
-    name: "lt_features"
-    data_type: TYPE_FP32
-    dims: [ 19, -1 ]
-  },
-  {
-    name: "lt_mask"
-    data_type: TYPE_FP32
-    dims: [ 1, -1 ]
+optimization {
+  graph {
+    level: -1
   }
-]
-output [
-  {
-    name: "output"
-    data_type: TYPE_FP32
-    dims: [ 6 ]
-    label_filename: "particlenet_labels.txt"
-  }
-]
-optimization {graph : {level : -1}}
+}
+model_repository_agents {
+  agents [{
+    name: "checksum"
+    parameters {
+      key: "MD5:1/model.onnx"
+      value: "6d362c09e8aafa399edf1226cee9d946"
+    }
+  }]
+}