Merge pull request #7 from kpedro88/ModelVersioning

add model checksums
cms-data · Jan 24, 2024 · 2c6273a · 2c6273a
2 parents da3357d + 882a119
commit 2c6273a
Show file tree

Hide file tree

Showing 2 changed files with 84 additions and 72 deletions.
diff --git a/models/deepmet/config.pbtxt b/models/deepmet/config.pbtxt
@@ -1,41 +1,49 @@
-# deepmet has two versions:
-# 1 is for the resolution tune
-# 2 is for the response tune
-
 name: "deepmet"
 platform: "tensorflow_graphdef"
-max_batch_size: 100
-input [
-  {
-    name: "input"
-    data_type: TYPE_FP32
-    dims: [ 4500, 8 ]
-  },
-  {
-    name: "input_cat0"
-    data_type: TYPE_FP32
-    dims: [ 4500, 1 ]
-  },
-  {
-    name: "input_cat1"
-    data_type: TYPE_FP32
-    dims: [ 4500, 1 ]
-  },
-  {
-    name: "input_cat2"
-    data_type: TYPE_FP32
-    dims: [ 4500, 1 ]
-  }
-]
-output [
-  {
-    name: "output/BiasAdd"
-    data_type: TYPE_FP32
-    dims: [ 2 ]
+version_policy {
+  all {
   }
-]
-
-version_policy: { all { }}
+}
+max_batch_size: 100
+input [{
+  name: "input"
+  data_type: TYPE_FP32
+  dims: [4500, 8]
+}, {
+  name: "input_cat0"
+  data_type: TYPE_FP32
+  dims: [4500, 1]
+}, {
+  name: "input_cat1"
+  data_type: TYPE_FP32
+  dims: [4500, 1]
+}, {
+  name: "input_cat2"
+  data_type: TYPE_FP32
+  dims: [4500, 1]
+}]
+output [{
+  name: "output/BiasAdd"
+  data_type: TYPE_FP32
+  dims: [2]
+}]
 dynamic_batching {
-    preferred_batch_size: [ 8, 16 ]
+  preferred_batch_size: [8, 16]
+}
+model_repository_agents {
+  agents [{
+    name: "checksum"
+    parameters {
+      key: "MD5:1/model.graphdef"
+      value: "9fdc487afe835467983c0bb0b2d50c3f"
+    }
+    parameters {
+      key: "MD5:2/model.graphdef"
+      value: "97b53089661709e24c7407d129d24e2e"
+    }
+    parameters {
+      key: "MD5:3/model.graphdef"
+      value: "111604a59f237dd49756fbfed24990db"
+    }
+  }]
 }
diff --git a/models/deepmet_phase2/config.pbtxt b/models/deepmet_phase2/config.pbtxt
@@ -1,41 +1,45 @@
-# deepmet has two versions:
-# 1 is for the resolution tune
-# 2 is for the response tune
-
 name: "deepmet_phase2"
 platform: "tensorflow_graphdef"
-max_batch_size: 100
-input [
-  {
-    name: "input"
-    data_type: TYPE_FP32
-    dims: [ 12500, 8 ]
-  },
-  {
-    name: "input_cat0"
-    data_type: TYPE_FP32
-    dims: [ 12500, 1 ]
-  },
-  {
-    name: "input_cat1"
-    data_type: TYPE_FP32
-    dims: [ 12500, 1 ]
-  },
-  {
-    name: "input_cat2"
-    data_type: TYPE_FP32
-    dims: [ 12500, 1 ]
-  }
-]
-output [
-  {
-    name: "output/BiasAdd"
-    data_type: TYPE_FP32
-    dims: [ 2 ]
+version_policy {
+  all {
   }
-]
-
-version_policy: { all { }}
+}
+max_batch_size: 100
+input [{
+  name: "input"
+  data_type: TYPE_FP32
+  dims: [12500, 8]
+}, {
+  name: "input_cat0"
+  data_type: TYPE_FP32
+  dims: [12500, 1]
+}, {
+  name: "input_cat1"
+  data_type: TYPE_FP32
+  dims: [12500, 1]
+}, {
+  name: "input_cat2"
+  data_type: TYPE_FP32
+  dims: [12500, 1]
+}]
+output [{
+  name: "output/BiasAdd"
+  data_type: TYPE_FP32
+  dims: [2]
+}]
 dynamic_batching {
-    preferred_batch_size: [ 8, 16 ]
+  preferred_batch_size: [8, 16]
+}
+model_repository_agents {
+  agents [{
+    name: "checksum"
+    parameters {
+      key: "MD5:1/model.graphdef"
+      value: "e9447af8b98a673b799842d054111c36"
+    }
+    parameters {
+      key: "MD5:2/model.graphdef"
+      value: "4c189e3602794baee4bb262509523ef4"
+    }
+  }]
 }