diff --git a/models/deepmet/config.pbtxt b/models/deepmet/config.pbtxt index e4b126d..1ac9bdf 100644 --- 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 index eddb099..79f2f56 100644 --- 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" + } + }] }