diff --git a/models/higgsInteractionNet/config.pbtxt b/models/higgsInteractionNet/config.pbtxt index a6de55f..6ace8d3 100644 --- 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 index 24ba730..0f6fda4 100644 --- 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 index 5249f7a..864de51 100644 --- 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" + } + }] +} diff --git a/models/particleNetFromMiniAODAK4PuppiCentral/config.pbtxt b/models/particleNetFromMiniAODAK4PuppiCentral/config.pbtxt index 4b08acf..176ad47 100644 --- a/models/particleNetFromMiniAODAK4PuppiCentral/config.pbtxt +++ b/models/particleNetFromMiniAODAK4PuppiCentral/config.pbtxt @@ -1,62 +1,63 @@ name: "particleNetFromMiniAODAK4PuppiCentral" 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: "01ff8c5a8bef9f67f3f999fae6136d10" + } + }] +} diff --git a/models/particleNetFromMiniAODAK4PuppiForward/config.pbtxt b/models/particleNetFromMiniAODAK4PuppiForward/config.pbtxt index f2a8fe3..cc100a3 100644 --- a/models/particleNetFromMiniAODAK4PuppiForward/config.pbtxt +++ b/models/particleNetFromMiniAODAK4PuppiForward/config.pbtxt @@ -1,62 +1,63 @@ name: "particleNetFromMiniAODAK4PuppiForward" 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: "3aaf54a48234ceed80c0c711c0bd3142" + } + }] +} diff --git a/models/particleNetFromMiniAODAK8/config.pbtxt b/models/particleNetFromMiniAODAK8/config.pbtxt index 9ad1597..da6d0d6 100644 --- a/models/particleNetFromMiniAODAK8/config.pbtxt +++ b/models/particleNetFromMiniAODAK8/config.pbtxt @@ -1,47 +1,51 @@ name: "particleNetFromMiniAODAK8" 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: [32, -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] +}] +output [{ + name: "output" + data_type: TYPE_FP32 + dims: [11] + 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: [ 32, -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 ] +optimization { + graph { + level: -1 } -] -output [ - { - name: "output" - data_type: TYPE_FP32 - dims: [ 11 ] - label_filename: "particlenet_labels.txt" - } -] -optimization {graph : {level : -1}} +} +model_repository_agents { + agents [{ + name: "checksum" + parameters { + key: "MD5:1/model.onnx" + value: "6e6b3ba631494c9cb210da28ff835b4a" + } + }] +} diff --git a/models/particlenet/config.pbtxt b/models/particlenet/config.pbtxt index 848fa4a..be79fcb 100644 --- a/models/particlenet/config.pbtxt +++ b/models/particlenet/config.pbtxt @@ -1,47 +1,51 @@ name: "particlenet" platform: "onnxruntime_onnx" -max_batch_size : 160 +max_batch_size: 160 +input [{ + name: "pf_points" + data_type: TYPE_FP32 + dims: [2, -1] +}, { + name: "pf_features" + data_type: TYPE_FP32 + dims: [25, -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: [11, -1] +}, { + name: "sv_mask" + data_type: TYPE_FP32 + dims: [1, -1] +}] +output [{ + name: "softmax" + data_type: TYPE_FP32 + dims: [20] + label_filename: "particlenet_labels.txt" +}] dynamic_batching { - preferred_batch_size: [ 80 ] + preferred_batch_size: [80] } -input [ - { - name: "pf_points" - data_type: TYPE_FP32 - dims: [ 2, -1 ] - }, - { - name: "pf_features" - data_type: TYPE_FP32 - dims: [ 25, -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: [ 11, -1 ] - }, - { - name: "sv_mask" - data_type: TYPE_FP32 - dims: [ 1, -1 ] +optimization { + graph { + level: -1 } -] -output [ - { - name: "softmax" - 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: "3bff150e279b1c942ea38e946135069c" + } + }] +} diff --git a/models/particlenet_AK4/config.pbtxt b/models/particlenet_AK4/config.pbtxt index ded4c0a..b111900 100644 --- a/models/particlenet_AK4/config.pbtxt +++ b/models/particlenet_AK4/config.pbtxt @@ -1,47 +1,51 @@ name: "particlenet_AK4" 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: [20, -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: [11, -1] +}, { + name: "sv_mask" + data_type: TYPE_FP32 + dims: [1, -1] +}] +output [{ + name: "softmax" + data_type: TYPE_FP32 + dims: [8] + 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: [ 20, -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: [ 11, -1 ] - }, - { - name: "sv_mask" - data_type: TYPE_FP32 - dims: [ 1, -1 ] +optimization { + graph { + level: -1 } -] -output [ - { - name: "softmax" - data_type: TYPE_FP32 - dims: [ 8 ] - label_filename: "particlenet_labels.txt" - } -] -optimization {graph : {level : -1}} +} +model_repository_agents { + agents [{ + name: "checksum" + parameters { + key: "MD5:1/model.onnx" + value: "bfca43d9c9694c2af49f1b596cfe5da0" + } + }] +} diff --git a/models/particlenet_AK4_PT/config.pbtxt b/models/particlenet_AK4_PT/config.pbtxt index 5e11764..b75b111 100644 --- a/models/particlenet_AK4_PT/config.pbtxt +++ b/models/particlenet_AK4_PT/config.pbtxt @@ -1,47 +1,51 @@ name: "particlenet_AK4_PT" platform: "pytorch_libtorch" -max_batch_size : 500 +max_batch_size: 500 +input [{ + name: "pf_points__0" + data_type: TYPE_FP32 + dims: [2, -1] +}, { + name: "pf_features__1" + data_type: TYPE_FP32 + dims: [20, -1] +}, { + name: "pf_mask__2" + data_type: TYPE_FP32 + dims: [1, -1] +}, { + name: "sv_points__3" + data_type: TYPE_FP32 + dims: [2, -1] +}, { + name: "sv_features__4" + data_type: TYPE_FP32 + dims: [11, -1] +}, { + name: "sv_mask__5" + data_type: TYPE_FP32 + dims: [1, -1] +}] +output [{ + name: "softmax__0" + data_type: TYPE_FP32 + dims: [8] + label_filename: "particlenet_labels.txt" +}] dynamic_batching { - preferred_batch_size: [ 200 ] + preferred_batch_size: [200] } -input [ - { - name: "pf_points__0" - data_type: TYPE_FP32 - dims: [ 2, -1 ] - }, - { - name: "pf_features__1" - data_type: TYPE_FP32 - dims: [ 20, -1 ] - }, - { - name: "pf_mask__2" - data_type: TYPE_FP32 - dims: [ 1, -1 ] - }, - { - name: "sv_points__3" - data_type: TYPE_FP32 - dims: [ 2, -1 ] - }, - { - name: "sv_features__4" - data_type: TYPE_FP32 - dims: [ 11, -1 ] - }, - { - name: "sv_mask__5" - data_type: TYPE_FP32 - dims: [ 1, -1 ] +optimization { + graph { + level: -1 } -] -output [ - { - name: "softmax__0" - data_type: TYPE_FP32 - dims: [ 8 ] - label_filename: "particlenet_labels.txt" - } -] -optimization {graph : {level : -1}} +} +model_repository_agents { + agents [{ + name: "checksum" + parameters { + key: "MD5:1/model.pt" + value: "bc28391a0850bb10556bcd67f3156075" + } + }] +} diff --git a/models/particlenet_AK8_MD-2prong/config.pbtxt b/models/particlenet_AK8_MD-2prong/config.pbtxt index bd6d6ee..f7bf5c5 100644 --- a/models/particlenet_AK8_MD-2prong/config.pbtxt +++ b/models/particlenet_AK8_MD-2prong/config.pbtxt @@ -1,47 +1,51 @@ name: "particlenet_AK8_MD-2prong" platform: "onnxruntime_onnx" -max_batch_size : 160 +max_batch_size: 160 +input [{ + name: "pf_points" + data_type: TYPE_FP32 + dims: [2, -1] +}, { + name: "pf_features" + data_type: TYPE_FP32 + dims: [20, -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: [11, -1] +}, { + name: "sv_mask" + data_type: TYPE_FP32 + dims: [1, -1] +}] +output [{ + name: "softmax" + data_type: TYPE_FP32 + dims: [8] + label_filename: "particlenet_labels.txt" +}] dynamic_batching { - preferred_batch_size: [ 128 ] + preferred_batch_size: [128] } -input [ - { - name: "pf_points" - data_type: TYPE_FP32 - dims: [ 2, -1 ] - }, - { - name: "pf_features" - data_type: TYPE_FP32 - dims: [ 20, -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: [ 11, -1 ] - }, - { - name: "sv_mask" - data_type: TYPE_FP32 - dims: [ 1, -1 ] +optimization { + graph { + level: -1 } -] -output [ - { - name: "softmax" - data_type: TYPE_FP32 - dims: [ 8 ] - label_filename: "particlenet_labels.txt" - } -] -optimization {graph : {level : -1}} +} +model_repository_agents { + agents [{ + name: "checksum" + parameters { + key: "MD5:1/model.onnx" + value: "a8647c821962d5b687dfbdbdca60525e" + } + }] +} diff --git a/models/particlenet_AK8_MD-2prong_PT/config.pbtxt b/models/particlenet_AK8_MD-2prong_PT/config.pbtxt index 8fd3051..67dcc4c 100644 --- a/models/particlenet_AK8_MD-2prong_PT/config.pbtxt +++ b/models/particlenet_AK8_MD-2prong_PT/config.pbtxt @@ -1,47 +1,51 @@ name: "particlenet_AK8_MD-2prong_PT" platform: "pytorch_libtorch" -max_batch_size : 160 +max_batch_size: 160 +input [{ + name: "pf_points__0" + data_type: TYPE_FP32 + dims: [2, -1] +}, { + name: "pf_features__1" + data_type: TYPE_FP32 + dims: [20, -1] +}, { + name: "pf_mask__2" + data_type: TYPE_FP32 + dims: [1, -1] +}, { + name: "sv_points__3" + data_type: TYPE_FP32 + dims: [2, -1] +}, { + name: "sv_features__4" + data_type: TYPE_FP32 + dims: [11, -1] +}, { + name: "sv_mask__5" + data_type: TYPE_FP32 + dims: [1, -1] +}] +output [{ + name: "softmax__0" + data_type: TYPE_FP32 + dims: [8] + label_filename: "particlenet_labels.txt" +}] dynamic_batching { - preferred_batch_size: [ 128 ] + preferred_batch_size: [128] } -input [ - { - name: "pf_points__0" - data_type: TYPE_FP32 - dims: [ 2, -1 ] - }, - { - name: "pf_features__1" - data_type: TYPE_FP32 - dims: [ 20, -1 ] - }, - { - name: "pf_mask__2" - data_type: TYPE_FP32 - dims: [ 1, -1 ] - }, - { - name: "sv_points__3" - data_type: TYPE_FP32 - dims: [ 2, -1 ] - }, - { - name: "sv_features__4" - data_type: TYPE_FP32 - dims: [ 11, -1 ] - }, - { - name: "sv_mask__5" - data_type: TYPE_FP32 - dims: [ 1, -1 ] +optimization { + graph { + level: -1 } -] -output [ - { - name: "softmax__0" - data_type: TYPE_FP32 - dims: [ 8 ] - label_filename: "particlenet_labels.txt" - } -] -optimization {graph : {level : -1}} +} +model_repository_agents { + agents [{ + name: "checksum" + parameters { + key: "MD5:1/model.pt" + value: "ae893c44b098b32f6b3fa8c43e3c635a" + } + }] +} diff --git a/models/particlenet_AK8_MassRegression/config.pbtxt b/models/particlenet_AK8_MassRegression/config.pbtxt index 02743fb..3fe4af8 100644 --- a/models/particlenet_AK8_MassRegression/config.pbtxt +++ b/models/particlenet_AK8_MassRegression/config.pbtxt @@ -1,47 +1,51 @@ name: "particlenet_AK8_MassRegression" platform: "onnxruntime_onnx" -max_batch_size : 200 +max_batch_size: 200 +input [{ + name: "pf_points" + data_type: TYPE_FP32 + dims: [2, -1] +}, { + name: "pf_features" + data_type: TYPE_FP32 + dims: [25, -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: [11, -1] +}, { + name: "sv_mask" + data_type: TYPE_FP32 + dims: [1, -1] +}] +output [{ + name: "output" + data_type: TYPE_FP32 + dims: [1] + 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: [ 25, -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: [ 11, -1 ] - }, - { - name: "sv_mask" - data_type: TYPE_FP32 - dims: [ 1, -1 ] +optimization { + graph { + level: -1 } -] -output [ - { - name: "output" - data_type: TYPE_FP32 - dims: [ 1 ] - label_filename: "particlenet_labels.txt" - } -] -optimization {graph : {level : -1}} +} +model_repository_agents { + agents [{ + name: "checksum" + parameters { + key: "MD5:1/model.onnx" + value: "a52c6aa60dbb9d926e9b335158ad9278" + } + }] +} diff --git a/models/particlenet_AK8_MassRegression_PT/config.pbtxt b/models/particlenet_AK8_MassRegression_PT/config.pbtxt index e0fa019..805ebf4 100644 --- a/models/particlenet_AK8_MassRegression_PT/config.pbtxt +++ b/models/particlenet_AK8_MassRegression_PT/config.pbtxt @@ -1,47 +1,51 @@ name: "particlenet_AK8_MassRegression_PT" platform: "pytorch_libtorch" -max_batch_size : 200 +max_batch_size: 200 +input [{ + name: "pf_points__0" + data_type: TYPE_FP32 + dims: [2, -1] +}, { + name: "pf_features__1" + data_type: TYPE_FP32 + dims: [25, -1] +}, { + name: "pf_mask__2" + data_type: TYPE_FP32 + dims: [1, -1] +}, { + name: "sv_points__3" + data_type: TYPE_FP32 + dims: [2, -1] +}, { + name: "sv_features__4" + data_type: TYPE_FP32 + dims: [11, -1] +}, { + name: "sv_mask__5" + data_type: TYPE_FP32 + dims: [1, -1] +}] +output [{ + name: "output__0" + data_type: TYPE_FP32 + dims: [1] + label_filename: "particlenet_labels.txt" +}] dynamic_batching { - preferred_batch_size: [ 200 ] + preferred_batch_size: [200] } -input [ - { - name: "pf_points__0" - data_type: TYPE_FP32 - dims: [ 2, -1 ] - }, - { - name: "pf_features__1" - data_type: TYPE_FP32 - dims: [ 25, -1 ] - }, - { - name: "pf_mask__2" - data_type: TYPE_FP32 - dims: [ 1, -1 ] - }, - { - name: "sv_points__3" - data_type: TYPE_FP32 - dims: [ 2, -1 ] - }, - { - name: "sv_features__4" - data_type: TYPE_FP32 - dims: [ 11, -1 ] - }, - { - name: "sv_mask__5" - data_type: TYPE_FP32 - dims: [ 1, -1 ] +optimization { + graph { + level: -1 } -] -output [ - { - name: "output__0" - data_type: TYPE_FP32 - dims: [ 1 ] - label_filename: "particlenet_labels.txt" - } -] -optimization {graph : {level : -1}} +} +model_repository_agents { + agents [{ + name: "checksum" + parameters { + key: "MD5:1/model.pt" + value: "23ca7010720c682ead36c08d2d730b6a" + } + }] +} diff --git a/models/particlenet_PT/config.pbtxt b/models/particlenet_PT/config.pbtxt index 0dfe6f5..59780a5 100644 --- a/models/particlenet_PT/config.pbtxt +++ b/models/particlenet_PT/config.pbtxt @@ -1,47 +1,51 @@ name: "particlenet_PT" platform: "pytorch_libtorch" -max_batch_size : 160 +max_batch_size: 160 +input [{ + name: "pf_points__0" + data_type: TYPE_FP32 + dims: [2, -1] +}, { + name: "pf_features__1" + data_type: TYPE_FP32 + dims: [25, -1] +}, { + name: "pf_mask__2" + data_type: TYPE_FP32 + dims: [1, -1] +}, { + name: "sv_points__3" + data_type: TYPE_FP32 + dims: [2, -1] +}, { + name: "sv_features__4" + data_type: TYPE_FP32 + dims: [11, -1] +}, { + name: "sv_mask__5" + data_type: TYPE_FP32 + dims: [1, -1] +}] +output [{ + name: "softmax__0" + data_type: TYPE_FP32 + dims: [20] + label_filename: "particlenet_labels.txt" +}] dynamic_batching { - preferred_batch_size: [ 80 ] + preferred_batch_size: [80] } -input [ - { - name: "pf_points__0" - data_type: TYPE_FP32 - dims: [ 2, -1 ] - }, - { - name: "pf_features__1" - data_type: TYPE_FP32 - dims: [ 25, -1 ] - }, - { - name: "pf_mask__2" - data_type: TYPE_FP32 - dims: [ 1, -1 ] - }, - { - name: "sv_points__3" - data_type: TYPE_FP32 - dims: [ 2, -1 ] - }, - { - name: "sv_features__4" - data_type: TYPE_FP32 - dims: [ 11, -1 ] - }, - { - name: "sv_mask__5" - data_type: TYPE_FP32 - dims: [ 1, -1 ] +optimization { + graph { + level: -1 } -] -output [ - { - name: "softmax__0" - 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.pt" + value: "a5b29f593329a6ecc67991b711ad7d07" + } + }] +}