Merge pull request #10 from yongbinfeng/mlpf

minor fix on the MLPF onnx model and upate the MLPF tensorflow model
fastmachinelearning · Nov 10, 2021 · e957909 · e957909
2 parents b30d828 + 9877cfa
commit e957909
Show file tree

Hide file tree

Showing 6 changed files with 54 additions and 143 deletions.
diff --git a/models/mlpf_acat2021/config.pbtxt b/models/mlpf_acat2021/config.pbtxt
@@ -1,21 +1,20 @@
 name: "mlpf_acat2021"
 platform: "onnxruntime_onnx"
-max_batch_size : 16
+max_batch_size : 200
 dynamic_batching {
-   preferred_batch_size: [ 8 ]
 }
 input [
   {
-    name: "x"
+    name: "x:0"
     data_type: TYPE_FP32
     dims: [ -1, 25 ]
   }
 ]
 output [
   {
-    name: "Identity"
+    name: "Identity:0"
     data_type: TYPE_FP32
     dims: [ -1, 14 ]
     label_filename: "mlpf_labels.txt"
   }
-]
+]
diff --git a/...lpf_acat2021_tf/model_full/saved_model.pb → ...2021_tf/1/model.savedmodel/saved_model.pb b/...lpf_acat2021_tf/model_full/saved_model.pb → ...2021_tf/1/model.savedmodel/saved_model.pb
diff --git a/...l/variables/variables.data-00000-of-00001 → ...l/variables/variables.data-00000-of-00001 b/...l/variables/variables.data-00000-of-00001 → ...l/variables/variables.data-00000-of-00001
diff --git a/...1_tf/model_full/variables/variables.index → ...odel.savedmodel/variables/variables.index b/...1_tf/model_full/variables/variables.index → ...odel.savedmodel/variables/variables.index
diff --git a/models/mlpf_acat2021_tf/config.pbtxt b/models/mlpf_acat2021_tf/config.pbtxt
@@ -0,0 +1,50 @@
+name: "mlpf_acat2021_tf",
+platform: "tensorflow_savedmodel",
+backend: "tensorflow",
+max_batch_size : 200
+dynamic_batching {
+}
+input: [
+    {
+        name: "input_1",
+        data_type: TYPE_FP32,
+        dims: [ -1, 25 ]
+    }
+]
+output: [
+    {
+        name: "cls",
+        data_type: TYPE_FP32,
+        dims: [ -1, 8 ]
+    },
+    {
+        name: "sin_phi",
+        data_type: TYPE_FP32,
+        dims: [ -1, 1 ]
+    },
+    {
+        name: "energy",
+        data_type: TYPE_FP32,
+        dims: [ -1, 1 ]
+    },
+    {
+        name: "pt",
+        data_type: TYPE_FP32,
+        dims: [ -1, 1 ]
+    },
+    {
+        name: "eta",
+        data_type: TYPE_FP32,
+        dims: [ -1, 1 ]
+    },
+    {
+        name: "charge",
+        data_type: TYPE_FP32,
+        dims: [ -1, 1 ]
+    },
+    {
+        name: "cos_phi",
+        data_type: TYPE_FP32,
+        dims: [ -1, 1 ]
+    }
+]